创建scikit学习对象

时间:2017-12-16 21:31:12

标签: python scikit-learn

想要创建像:

这样的对象
regr = linear_model.LinearRegression()

with:

GaussianNB()
SVC() 
svm.SVR()
DecisionTreeClassifier()

例如,对于:

from sklearn.naive_bayes import GaussianNB
gaussian = naive_bayes.GaussianNB()

我明白了:

NameError: name 'naive_bayes' is not defined

有关:

from  sklearn.tree import DecisionTreeClassifier
treeclassifier = tree.DecisionTreeClassifier()

我明白了:

NameError: name 'tree' is not defined

提前致谢

2 个答案:

答案 0 :(得分:2)

从子模块导入对象时,可以直接使用该对象:例如

from sklearn.naive_bayes import GaussianNB
model = GaussianNB()

导入子模块本身时,可以在代码中使用该子模块名称。例如:

from sklearn import naive_bayes
model = naive_bayes.GaussianNB()

确保您的import语句符合您的代码!

答案 1 :(得分:0)

对于每一个,都有一个API页面,如this one。每个API页面至少提供一个示例(基本用法)!

如果您发现错误,可能是由于导入风格。我目前找不到一些文档页面,但基本的例子都是你需要的。

所以它只是:

from sklearn.naive_bayes import GaussianNB
clf = GaussianNB()

from sklearn.svm import SVC
clf = SVC()

from sklearn.svm import SVR
clf = SVR()

from sklearn.tree import DecisionTreeClassifier
clf = DecisionTreeClassifier()