python - 如何在sklearn中计算.fit()训练模型的实际大小？

Question

是否可以在 scikit-learn 中计算模型(假设是随机森林分类器)的大小？

例如:

  from sklearn.ensemble import RandomForestClassifier
  clf = RandomForestClassifier(n_jobs=-1, n_estimators=10000, min_samples_leaf=50)
  clf.fit(self.X_train, self.y_train)

我可以确定clf的大小吗？

Answer 1

与 Nijan 的答案相同，您也可以使用 pickle 来完成此操作，而无需保存模型:

import pickle
import sys

p = pickle.dumps(clf)
print(sys.getsizeof(p))

它将返回以字节为单位的大小。