Keras AttributeError: 'History' object has no attribute 'predict'
Note: I have seen this related post but I don't know I can use
the answer for my problem.
我尝试使用Keras进行简单回归。为此,我创建了一个简单的
1 2 3 4 5 6 7 8 9 10 11 | def policy_network(): model = Sequential() model.add(MaxPooling2D(pool_size=(4, 4),input_shape=[64,64,3])) model.add(Flatten()) model.add(Dense(1, kernel_initializer='normal', activation='linear')) model.compile(loss='mean_squared_error', optimizer=Adam(lr=learning_rate), metrics=['mean_squared_error']) return model |
我还定义了一个全局变量
1 2 3 4 | policy_network = policy_network().fit(images, actions, batch_size=256, epochs=10, shuffle=True) |
但是当我打电话给
时
1 | action = policy_network.predict(image) |
我得到了
Keras的
这样做:
1 2 3 4 5 6 | model = policy_network() model.fit(images, actions, batch_size=256, epochs=10, shuffle=True) action = model.predict(image) |
对Python说时,您已将policy_network的类从keras.Model对象更改为History对象。
1 | policy_network = policy_network().fit(..) |
如果要将历史记录存储在变量中,请将其存储在另一个变量中:
1 | history = policy_network.fit(..) |
您现在可以按照自己的方式使用