在另一个代码中使用scikit SVM生成的模型

Question

我在名为main.py的python脚本中训练了一个LinearSVC，用于训练图像分类算法。模型看起来像这样。

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

我需要使用这个训练过的模型来预测另一个代码中的图像类。如何将生成的模型（即lin_clf）导出到其他代码。

提前谢谢。

Answer 1

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

这是一个示例，说明如何使用shelve：

将lin_clf对象写入磁盘

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