python - 在另一个代码中使用 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()