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

2024-06-08 20:54:05 发布

您现在位置:Python中文网/ 问答频道 /正文

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

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

如果我使用不同的空闲时间会更好吗


Tags: https模型脚本com编码step神经网络卷积
1条回答
网友
1楼 · 发布于 2024-06-08 20:54:05

您可以使用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) 

相关问题 更多 >