【python】【codelet】scikit-learn几个模型

最新推荐文章于 2024-09-17 23:15:58 发布

无意1975

最新推荐文章于 2024-09-17 23:15:58 发布

阅读量873

点赞数

分类专栏： python 文章标签： python scikit-learn 支持向量机

本文链接：https://blog.csdn.net/GreatJinYun/article/details/121923013

版权

python 专栏收录该内容

10 篇文章 0 订阅

订阅专栏


# Step 2）模型构建
# -----------------------------------------------------
strModelName = "JinModel";
from sklearn.linear_model import RidgeClassifier
from sklearn import tree;
from sklearn import svm;
from sklearn.pipeline import make_pipeline;
from sklearn.preprocessing import StandardScaler;
from sklearn.neural_network import MLPClassifier;
from sklearn.neighbors import KNeighborsClassifier;

if args.model=='ridge':
    # 算法01） 
    strModelName = "RidgeClassifier";
    clf = RidgeClassifier();
elif args.model=='decision':
    # 算法02） 决策树
    strModelName = "DecisionTreeClassifier";
    clf = tree.DecisionTreeClassifier();
elif args.model=='linearsvc':
    # 算法03）支持向量机
    strModelName = "LinearSVC";
    clf = svm.LinearSVC( C=0.8 );
elif args.model=='stsvc':
    # 算法04）支持向量机，标准化数据
    strModelName = "StScalerSVC";
    clf = make_pipeline(StandardScaler(), svm.SVC(kernel='rbf',  C=4, gamma=0.15 ));
elif args.model=='nusvc':
    # 算法06）非线性支持向量机，标准化数据
    strModelName = "NuSVC";
    clf = make_pipeline(StandardScaler(), svm.NuSVC(gamma='auto' ));
elif args.model=='mlp':
    # 算法07）神经网络训练
    strModelName = "MLPClassifier";
    clf = make_pipeline(StandardScaler(), MLPClassifier(solver='lbfgs', alpha=3,hidden_layer_sizes=(40, ), random_state=1, max_iter=1000)  );
elif args.model=='nearest':
    # 算法08）最近邻算法
    strModelName = "NearestNeighbor";
    n_neighbors = 15;
    weights = ["uniform", "distance"];
    clf = make_pipeline(StandardScaler(), KNeighborsClassifier(n_neighbors, weights=weights[1]) );
else:
    #
    strModelName = "RidgeClassifier";
    clf = RidgeClassifier();
# =======================================================


print("\n----------- "+strModelName+" ------------");