基于KNN算法对不同类型的充电站进行识别优化

 import lightgbm as lgb

    
 import pandas as pd
    
 import numpy as np
    
 from sklearn.model_selection import train_test_split
    
 # 导入数据
    
  
    
 col_names = ["ID","K1K2驱动信号","电子锁驱动信号","急停信号","门禁信号","THDV-M","THDI-M","label"]
    
 data = pd.read_csv("data_train.csv",names=col_names)
    
 dataset_X = data[["K1K2驱动信号","电子锁驱动信号","急停信号","门禁信号","THDV-M","THDI-M"]].as_matrix()
    
 dataset_Y = data[["label"]].as_matrix()
    
 dataset_Y = np.array(dataset_Y).reshape(len(dataset_Y))
    
  
    
 x_train, x_test, y_train, y_test = train_test_split(dataset_X, dataset_Y,test_size