DecisionTreeClassifier保存模型
时间: 2024-01-05 17:04:30 浏览: 167
二手车交易价格预测_Task5_模型融合
要保存DecisionTreeClassifier模型,可以使用Python中的pickle或joblib库。以下是使用joblib库保存和加载模型的示例代码:
```python
from sklearn.tree import DecisionTreeClassifier
from sklearn import datasets
from sklearn.externals import joblib
# 加载数据集
iris = datasets.load_iris()
X = iris.data
y = iris.target
# 训练模型
clf = DecisionTreeClassifier()
clf.fit(X, y)
# 保存模型
joblib.dump(clf, 'model.pkl')
# 加载模型
clf2 = joblib.load('model.pkl')
```
在这个例子中,我们使用了joblib.dump()将训练好的决策树分类器保存到名为'model.pkl'的文件中。然后,我们使用joblib.load()从文件中加载模型并将其存储在clf2变量中。
阅读全文