使用常用的机器学习算法对水果数据集进行分类（逻辑回归、决策树、KNN、SVM等）

数据集下载地址：

链接：https://pan.baidu.com/s/1vhVlalmKnvMB4sLLP2VXFg
提取码：zhrn

导入库函数

%matplotlib inline
import numpy as np
import pandas as pd
import matplotlib.pyplot as plt

读取数据集合

fruits = pd.read_table('fruit_data_with_colors.txt')
fruits.head()

 

查看水果类型

print(fruits['fruit_name'].unique())

['apple' 'mandarin' 'orange' 'lemon']

包含四种水果。

查看数据集的统计指标

fruits.describe()

 查看没类水果数据集的数量：