tensorflow - Keras 模型适合在 Jupyter 笔记本输出中创建正方形

Question

我将 Keras 2.0.2 与 TensorFlow 结合使用:

我正在运行一个简单的模型:

from keras.layers.core import Lambda, Flatten, Dense
from keras.models import Sequential
from keras.optimizers import Adam

model = Sequential([
    Lambda(norm_inp, input_shape=(1, 28, 28)),
    Flatten(),
    Dense(10, activation="softmax")
])

model.compile(optimizer=Adam(), loss='categorical_crossentropy', metrics=['accuracy'])

def run_fit(model, lr = 0.0001, epochs = 1):
    model.optimizer.lr = lr

    model.fit(x_train_processed, y_train_processed, epochs=epochs, verbose=1,
              validation_data=(x_test_processed, y_test_processed))

run_fit(model, 0.0001, 1)

但是当我在 Jupyter 中运行它时，我在输出中得到了正方形。

我尝试搜索此内容，但找不到任何相关内容。

Answer 1

尝试verbose=0或verbose=2。

您可能还想使用 keras-tqdm:https://github.com/bstriner/keras-tqdm