以下算法都是在sklearn库中。 (1)决策树 from sklearn import tree model = tree.DecisionTreeClassifier(criterion='entropy') (2)逻辑回归 from sklearn.linear_model import LogisticRegression model = LogisticRegression() (3)随机森林 from sklearn.ensemble import RandomForestRegressor model = RandomForestRegressor() (4)支持向量机 from sklearn import svm model = svm.SVC(kernel='poly', gamma=1) (5)k近邻 from sklearn.neighbors import KNeighborsClassifier model=KNeighborsClassifier(metric='minkowski')
训练模型:
model.fit(x_train, y_train)
print(model.score(x_train, y_train))
print(model.score(x_dev, y_dev))
预测数据:
pred = model.predict(data)