Pycharm:有没有一种方法可以在不运行整个文件的情况下运行一段代码?

时间:2020-11-05 18:55:31

标签: python pycharm cnn

我正在自学编码卷积神经网络。我特别关注“狗与猫”的挑战(https://medium.com/@mrgarg.rajat/kaggle-dogs-vs-cats-challenge-complete-step-by-step-guide-part-2-e9ee4967b9)。我正在使用PyCharm。

在PyCharm中,是否有一种方法可以使用训练有素的模型对测试数据进行预测,而不必每次都运行整个文件(因此每次都需要对模型进行重新训练)?另外,是否有一种方法可以跳过脚本中准备将数据输入到CNN中的部分? PyCharm是否以类似的方式存储变量-运行脚本后我是否可以打印单个变量。

如果我使用其他IDLE,会更好吗?

1 个答案:

答案 0 :(得分:0)

您可以使用sklearn joblib将经过训练的模型另存为pickle,以后再用于预测。

from sklearn.externals import joblib 

# Save the model as a pickle in a file 
joblib.dump(knn, 'filename.pkl') 

# Load the model from the file 
knn_from_joblib = joblib.load('filename.pkl')  

# Use the loaded model to make predictions 
knn_from_joblib.predict(X_test) 
相关问题