117.info
人生若只如初见

Scikit-learn中怎么实现K最近邻

在Scikit-learn中,可以使用KNeighborsClassifier类来实现K最近邻算法。以下是一个简单的示例代码:

from sklearn.neighbors import KNeighborsClassifier
from sklearn.datasets import load_iris
from sklearn.model_selection import train_test_split
from sklearn.metrics import accuracy_score

# 加载数据集
iris = load_iris()
X = iris.data
y = iris.target

# 划分数据集
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2, random_state=42)

# 创建K最近邻分类器
knn = KNeighborsClassifier(n_neighbors=3)

# 拟合模型
knn.fit(X_train, y_train)

# 预测
y_pred = knn.predict(X_test)

# 计算准确率
accuracy = accuracy_score(y_test, y_pred)
print("准确率:", accuracy)

在上面的代码中,首先加载了鸢尾花数据集,然后将数据集划分为训练集和测试集。接着创建了一个KNeighborsClassifier对象,并使用fit()方法拟合模型。最后使用predict()方法进行预测,并计算准确率。

未经允许不得转载 » 本文链接:https://www.117.info/ask/fe14bAzsIBwBfAlA.html

推荐文章

  • Scikit-learn中怎么实现线性回归

    在Scikit-learn中,可以使用LinearRegression类来实现线性回归。下面是一个简单的示例代码:
    from sklearn.linear_model import LinearRegression
    imp...

  • Scikit-learn中怎么使用特征提取

    在Scikit-learn中,特征提取通常是通过特征提取器(如CountVectorizer、TfidfVectorizer等)来实现的。下面是一个简单的使用CountVectorizer来提取文本特征的示例...

  • Scikit-learn中怎么使用特征选择

    在Scikit-learn中,可以使用特征选择技术通过sklearn.feature_selection模块中提供的方法来选择最重要的特征。下面是一个简单的例子来展示如何使用特征选择:

  • Scikit-learn中怎么使用数据归一化

    使用Scikit-learn中的数据归一化,可以使用preprocessing模块中的MinMaxScaler类来实现。具体步骤如下: 导入MinMaxScaler类: from sklearn.preprocessing impo...

  • Scikit-learn中怎么实现支持向量机

    在Scikit-learn中,支持向量机模型可以通过svm模块中的SVC类来实现。下面是一个简单的示例代码:
    from sklearn import svm
    from sklearn import datas...

  • Scikit-learn中怎么实现随机森林

    在Scikit-learn中,可以使用RandomForestClassifier类来实现随机森林模型。以下是一个简单的示例代码:
    from sklearn.ensemble import RandomForestClassif...

  • Scikit-learn中怎么实现决策树

    在Scikit-learn中实现决策树的方法是使用DecisionTreeClassifier类。下面是一个简单的示例代码,演示如何在Scikit-learn中实现决策树:
    from sklearn impor...

  • Scikit-learn中怎么实现逻辑回归

    要使用Scikit-learn实现逻辑回归,首先需要导入LogisticRegression模块,然后创建LogisticRegression对象,并调用fit方法拟合模型,最后使用predict方法进行预测...