serialize adaboost分类器scikit-learn

Question

我尝试使用cPickle来序列化输出分类器，以将其保存到数据库或文件中，但是出现内存不足错误，当我使用marshal时，它给了我unmarshable对象。所以，我想知道如何序列化这个学习过的分类器。

def adboost_classify(X,Y):
   bdt = AdaBoostClassifier(DecisionTreeClassifier(max_depth=10),
                    algorithm="SAMME.R",
                     n_estimators=3000)
   t0 = time()
   bdt.fit(X, Y)
   t1 = time()
   thebytes = cPickle.dumps(bdt)

提前谢谢你

Answer 1

这是因为您尝试将整个表示存储在内存中。尝试将其直接写入文件：

with open('adaboostpickled.tmp', 'w') as output:
    cPikle.dump(bdt, output)