在Jupyter notebook中，如何使用回调和/或小部件事件手动终止Keras培训？

import tensorflow as tf
from tensorflow.keras.callbacks import Callback
from ipywidgets import Button

class TerminateTrainingCallback(Callback):
    def on_train_end(self, logs=None):
        print('Training manually terminated.')

button = Button(description='Terminate Training')

def on_button_clicked(button):
    # 触发终止信号
    model.stop_training = True
    print('Termination signal sent.')

button.on_click(on_button_clicked)

model = tf.keras.Sequential([...])  # 创建模型，其中包含所需的层
model.compile(optimizer='adam', loss='sparse_categorical_crossentropy', metrics=['accuracy'])
model.callbacks = [TerminateTrainingCallback()]

model.fit(x_train, y_train, epochs=10, callbacks=model.callbacks)

display(button)