Skip to main content
 首页 » 编程设计

python之使用 Keras Python 测试神经网络

2025年05月04日24飞鱼

我已经使用 Python 中的 Keras 和数据集训练和测试了前馈神经网络。但是每次,为了识别具有外部数据的新测试集(外部数据不包含在数据集中),我都必须重新训练前馈神经网络来计算测试集。例如每次我必须做的:

 model.fit (data, output_data) 
 prediction=model.predict_classes(new_test) 
 print "Prediction : " prediction 

获得正确的输出:

  Prediction: [1 2 3 4 5 1 2 3 1 2 3] 
  Acc: 100% 

现在我将测试一个新的测试集,即“new_test2.csv”,而无需再次重新训练,只使用网络学到的东西。我也在考虑一种实时识别。

我应该怎么做?

提前致谢

请您参考如下方法:

借助训练有素的模型,您可以对任何新数据进行预测。你不必重新训练任何东西,因为(希望)你的模型可以将它的学习概括为看不见的数据,并将达到相当的准确性。

只需将“new_test2.csv”中的数据输入到您的预测函数中:

prediction=model.predict_classes(content_of_new_test2) 

显然,您需要相同类型和类别的数据。除此之外,您还需要对新数据应用任何转换,就像您可能已经转换训练模型所用的数据一样。

如果您想要实时预测,您可以使用 Flask 设置 API:

http://flask.pocoo.org/

关于术语和正确的训练方法:

您在训练集(例如您拥有的所有数据的 70%)上训练。

您使用验证集(例如 15% 的数据)验证您的训练。您使用训练中的准确性和损失值来调整您的超参数。

然后,您通过预测来自您的测试集的数据(同样是您的数据的 15%)来评估您的模型的最终性能。那必须是数据,你的网络以前根本没有见过,也没有被你用来优化训练参数。

之后,您可以预测生产数据。

如果您想保存经过训练的模型,请使用这个(取自 Keras 文档):

from keras.models import load_model 
 
model.save('my_model.h5')  # creates a HDF5 file 'my_model.h5' 
del model  # deletes the existing model 
 
# returns a compiled model 
# identical to the previous one 
model = load_model('my_model.h5')  

https://keras.io/getting-started/faq/#how-can-i-save-a-keras-model