当前位置:   article > 正文

机器学习之Scikit-learn基础教程_引入scikitlearn

引入scikitlearn

        Scikit-learn(简称sklearn)是一个广泛使用的Python机器学习库,它提供了各种算法和工具,用于数据挖掘和数据分析。本教程将介绍sklearn的基本概念和使用方法。

1. 安装Scikit-learn

如果你还没有安装scikit-learn,可以通过pip进行安装:

  1. bash
  2. pip install scikit-learn

2. 导入库和加载数据

在Python中,你可以这样导入scikit-learn和其他常用库:

  1. python
  2. import numpy as np
  3. import pandas as pd
  4. from sklearn import datasets

加载数据集通常很简单,scikit-learn自带了一些标准数据集,例如鸢尾花数据集:

  1. python
  2. iris = datasets.load_iris()
  3. X = iris.data
  4. y = iris.target

3. 数据预处理

数据预处理是机器学习中的重要步骤,scikit-learn提供了许多工具来帮助完成这项工作:

  1. python
  2. from sklearn.model_selection import train_test_split
  3. from sklearn.preprocessing import StandardScaler
  4. # 划分数据集
  5. X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2, random_state=42)
  6. # 特征缩放
  7. scaler = StandardScaler()
  8. X_train = scaler.fit_transform(X_train)
  9. X_test = scaler.transform(X_test)

4. 选择模型

scikit-learn提供了多种监督学习和非监督学习算法。以下是一些常用的模型:

  • 线性回归(LinearRegression)
  • 支持向量机(SVM,如SVC)
  • 决策树(DecisionTreeClassifier)
  • 随机森林(RandomForestClassifier)
  • K-最近邻(KNeighborsClassifier)选择模型并实例化:
  1. python
  2. from sklearn.ensemble import RandomForestClassifier
  3. model = RandomForestClassifier(n_estimators=100, random_state=42)

5. 训练模型

使用训练数据来训练(或称拟合)模型:

  1. python
  2. model.fit(X_train, y_train)

6. 模型评估

评估模型的性能,通常在测试集上进行:

  1. python
  2. from sklearn.metrics import accuracy_score
  3. y_pred = model.predict(X_test)
  4. accuracy = accuracy_score(y_test, y_pred)
  5. print(f"Model Accuracy: {accuracy:.2f}")

7. 模型优化

使用交叉验证、网格搜索等方法对模型进行优化:

  1. python
  2. from sklearn.model_selection import GridSearchCV
  3. param_grid = {
  4.     'n_estimators': [50, 100, 200],
  5.     'max_depth': [None, 10, 20, 30]
  6. }
  7. grid_search = GridSearchCV(estimator=model, param_grid=param_grid, cv=5)
  8. grid_search.fit(X_train, y_train)
  9. print(f"Best Parameters: {grid_search.best_params_}")
  10. print(f"Best CV Score: {grid_search.best_score_}")

8. 使用模型进行预测

  一旦模型被训练和优化,就可以用它来进行预测:

  1. python
  2. new_samples = np.array([[5.1, 3.5, 1.4, 0.2]])  # 新样本特征
  3. new_samples = scaler.transform(new_samples)  # 预处理新样本
  4. predictions = model.predict(new_samples)
  5. print(f"Predictions: {predictions}")

 

9. 结论

        Scikit-learn是一个功能强大且易于使用的机器学习库。本教程提供了一个基本的入门指南,帮助你开始使用scikit-learn进行机器学习项目。然而,机器学习是一个广阔的领域,还有许多高级主题等待探索,比如特征工程、模型选择、超参数调优等。

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/盐析白兔/article/detail/735281
推荐阅读
相关标签
  

闽ICP备14008679号