美国人口普查数据预测收入sklearn算法汇总3之ROC: KNN,LogisticRegression,RandomForest,NaiveBayes,StochasticGradientDescent

<深入解析美国人口普查数据及收入预测模型：基于sklearn算法的研究与实现>
<从特征工程到模型优化：美国人口普查收入预测算法系列研究（二）>

九. 机器学习算法

K nearest neighbors algorithm,
logistic regression model,
random forest algorithm,
naive Bayes classifier,
stochastic gradient descent algorithm,
linear support vector machine,
decision tree algorithm,
gradient boosting machine

    import random
    random.seed(42)
    
    from sklearn.neighbors import KNeighborsRegressor #K近邻
    knn = KNeighborsRegressor(n_neighbors = 3)
    knn.fit(X_train, y_train)
    print('KNN score: ',knn.score(X_test, y_test))
    
    from sklearn.linear_model imp