Parfois, nous créons un modèle, mais lorsque nous voulons utiliser ce modèle à plusieurs reprises, c'est plus compliqué. Nous devons écrire à plusieurs reprises ce modèle dans différents fichiers. À ce stade, pickle est très utile et peut nous aider. Le modèle créé est similaire au package, enregistré sous un autre fichier .pickle, puis lorsque nous voulons appeler ce modèle, il suffit de l'importer. Pour plus de détails sur son fonctionnement, consultez l'analyse suivante
Enregistrer le modèle
L'utilisation du code suivant générera automatiquement un fichier nommé model.pickle dans le répertoire courant, wb signifie que l'écriture est autorisée et est sous forme binaire
import pickle
with open("model.pickle", "wb") as f: # 保存模型
pickle.dump(model, f)
Utiliser le modèle
import pickle # 导入pickle用于接下来引用我们保存的模型
with open("model.pickle","rb") as f:
model = pickle.load(f)
test = [[10,16,6,22,0.62,53]] # 这里是用于预测的特征值,可以自己设置
y_predict = model.predict(test)
print(y_predict)