第八课 机器学习简单介绍
第13节 模型调参方法
# 引入包
import pandas as pd
import numpy as np
from sklearn.model_selection import train_test_split
准备数据集
# 加载数据
iris_data = pd.read_csv('./dataset/iris.csv')
# 获取特征
X = iris_data[['sepal_length', 'sepal_width', 'petal_length', 'petal_width']].values
# 获取标签
y = iris_data['label'].values
# 划分数据集
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=1/3, random_state=10)
建立模型
from sklearn.neighbors import KNei