KNN预测水果类型

代码

# -*- coding: utf-8 -*-
'''
   制作一个依据水果的尺寸和颜色判断其水果类型的knn算法  2019/02/03
'''

import  pandas  as pd
from  sklearn.model_selection  import  train_test_split
from  sklearn.neighbors import  KNeighborsClassifier
import random

FRUIT_DATA_FILE = './data/fruit_data.csv'

SPECIES_LABEL_DICT = {
   
    'apple' : 0,
    'mandarin' : 1,
    'orange' : 2 ,
    'lemon' : 3,
}

FEATURE_COL = ['mass','width','height','color_score']

def main():
    #  我原先输入的是： in