简单的Scikit-Learn入门示例

以下是一个简单的Scikit-Learn入门示例,该示例使用鸢尾花(Iris)数据集来演示分类问题的基本步骤。这个数据集包含了150个样本,每个样本有四个特征(花萼长度、花萼宽度、花瓣长度、花瓣宽度),以及一个目标变量(花的种类,分为山鸢尾、杂色鸢尾和维吉尼亚鸢尾)。

# 导入必要的库
from sklearn.datasets import load_iris
from sklearn.model_selection import train_test_split
from sklearn.preprocessing import StandardScaler
from sklearn.svm import SVC
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)

# 数据预处理(可选步骤,这里使用特征缩放)
scaler = StandardScaler()
X_train_scaled = scaler.fit_transform(X_train)
X_test_scaled = scaler.transform(X_test)

# 选择并训练模型(这里使用支持向量机SVM)
model = SVC(kernel='linear', C=1, random_state=42)
model.fit(X_train_scaled, y_train)

# 预测测试集结果
y_pred = model.predict(X_test_scaled)

# 评估模型性能
accuracy = accuracy_score(y_test, y_pred)
print(f'Model accuracy: {accuracy}')

在这个示例中,我们首先导入了必要的库,然后加载了鸢尾花数据集。接着,我们使用train_test_split函数将数据集划分为训练集和测试集。然后,我们选择了一个数据预处理方法(特征缩放),并用它处理了训练集和测试集的特征。之后,我们选择了支持向量机(SVM)作为分类器,并使用训练集数据训练了模型。最后,我们用模型对测试集进行了预测,并计算了模型的准确率。

相关推荐

  1. 简单Scikit-Learn入门示例

    2024-06-17 12:26:02       32 阅读
  2. Scikit-Learn认识

    2024-06-17 12:26:02       29 阅读
  3. Scikit Learn中支持单变量特征选择SVM示例

    2024-06-17 12:26:02       23 阅读
  4. scikit-learn保姆级入门教程

    2024-06-17 12:26:02       36 阅读

最近更新

  1. docker php8.1+nginx base 镜像 dockerfile 配置

    2024-06-17 12:26:02       94 阅读
  2. Could not load dynamic library ‘cudart64_100.dll‘

    2024-06-17 12:26:02       101 阅读
  3. 在Django里面运行非项目文件

    2024-06-17 12:26:02       82 阅读
  4. Python语言-面向对象

    2024-06-17 12:26:02       91 阅读

热门阅读

  1. 数组类模板(超详细)

    2024-06-17 12:26:02       27 阅读
  2. 使用Selenium进行元素定位的全面指南

    2024-06-17 12:26:02       21 阅读
  3. 安卓项目模块集成JNI的两种方式回顾

    2024-06-17 12:26:02       25 阅读
  4. 数据可视化:让数据讲述故事

    2024-06-17 12:26:02       32 阅读
  5. 使用 Verdaccio 建立私有npm库

    2024-06-17 12:26:02       33 阅读
  6. vue router中的导航守卫,它的5个使用场景举例

    2024-06-17 12:26:02       29 阅读
  7. SVG in VSCode: A Comprehensive Guide

    2024-06-17 12:26:02       32 阅读