学习《机器学习100天》第19天 决策树代码实现

github上的项目,跟着一起学习

项目地址    https://github.com/MLEveryday/100-Days-Of-ML-Code

这一天的任务:决策树的代码实现

导入需要的库:numpy、matplotlib.pyplot、pandas

导入数据集:pandas.read_csv()

将数据集拆分为训练集和数据集:从sklearn.model_selection导入train_test_split

特征缩放:从sklearn.preprocessing导入StandardScaler

对测试集进行决策树分类拟合:

   从sklearn.tree导入DecisionTreeClassifier

   调用fit

预测测试集结果:调用predict

可视化:绘图

猜你喜欢

转载自blog.csdn.net/a776995799/article/details/87259086
今日推荐