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

时间:2017-07-04 12:11:58

标签: tensorflow keras jupyter

我使用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中运行时,我在输出中得到了正方形。

enter image description here

我试图搜索这个,但我找不到任何相关内容。

1 个答案:

答案 0 :(得分:2)

尝试verbose=0verbose=2

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