在另一个cod中使用scikit-SVM生成模型

Question

我在一个名为主.py，用于训练图像分类算法。模型是这样的。你知道吗

lin_clf = svm.LinearSVC()
lin_clf.fit(feature,g)

我需要使用这个经过训练的模型来预测另一个代码中的图像类。如何将Generated模型（即lin\u clf）导出到其他代码。你知道吗

先谢谢你。你知道吗

Answer 1

我知道你的“其他代码”是另一个python脚本。 在这种情况下，您当然可以使用pickle或shelve模块将lin\u clf写入磁盘主.py，并从将使用该模型的脚本中的磁盘读取它。你知道吗

下面的示例演示如何使用shelve将lin\u clf对象写入磁盘：

import shelve
a = shelve.open("output")
a['lin_clf'] = lin_clf
a.close()