使用KNN实现充电桩分类

最新推荐文章于 2024-07-23 08:00:00 发布

王小波_Libo

最新推荐文章于 2024-07-23 08:00:00 发布

阅读量621

点赞数 1

分类专栏： # Python小项目文章标签：分类 KNN

本文链接：https://blog.csdn.net/qq_38900441/article/details/83472463

版权

这篇博客介绍了如何利用KNN（K近邻）算法对百度点石充电桩数据集进行分类，通过提供的源码实现了高精度的分类效果，准确率达到了100%。

摘要由CSDN通过智能技术生成

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=0.2,)

# create dataset for lightgbm
lgb_train = lgb.Dataset(x_train, y_train)
lgb_eval = lgb.Dataset(x_test, y_test, reference=lgb_train)


param = {
    'max_depth':8,
    'num_leaves':16,
    'learning_rate':0.4,
    'scale_p