sklearn里面比如定义一个RandomForestClassifier: clf

Question

sklearn里面比如定义一个RandomForesstClassifier: clf,那么执行clf.fit(X, y），model = clf.fit(X,y)之后，clf与model这个变量是一样的吗？

Answer 1

在代码中用print查看一下数据类型或值，就可以知道两者是否一样了。

from sklearn.ensemble import RandomForestClassifier
from sklearn.datasets import make_classification
X, y = make_classification(n_samples=1000, n_features=4,n_informative=2, n_redundant=0,random_state=0, shuffle=False)
clf = RandomForestClassifier(max_depth=2, random_state=0)
clf.fit(X, y)
model=clf.fit(X, y)
print(type(clf), clf)
print(type(model),model)
print(clf.feature_importances_)
print(clf.predict([[0, 0, 0, 0]]))