Sklearn神经网络训练和测试分两个步骤 - PYTHON

时间:2017-05-17 10:51:55

标签: python scikit-learn neural-network

我使用sklearn python创建了一个神经网络:

 mlp=MLPClassifier()
 mlp.fit(X_train,y_train)

我在python中运行代码并且NN已经过训练。

现在我会预测一个不同阶段的事情,然后不用这条线重新训练NN:

 prediction=mlp.predict(X_test)

基本上,拟合和预测调用应该在两个不同的步骤中完成,而不是每次重新训练NN,例如我将验证一组新数据:

 prediction2=mlp.predict(Z_test) 

1 个答案:

答案 0 :(得分:0)

您可以使用pickle将Python对象保存到磁盘。所以你有一个train.py pickle MLP,然后是test.py,你可以unpickle来自本地磁盘的对象。