#sklearn #随机森林

砸吧砸吧

于 2025-03-22 22:23:08 发布

阅读量103

点赞数 2

文章标签： python 机器学习人工智能

本文链接：https://blog.youkuaiyun.com/m0_49629753/article/details/146447780

版权

from sklearn.ensemble import RandomForestClassifier
from sklearn.metrics import accuracy_score
import pandas as pd


df = pd.read_csv("result.csv")

x_train = df.drop(columns=['status','device_id'])[:6]
y_train = df['status'][:6]

x_test = df.drop(columns=['status','device_id'])[6:]
y_test = df['status'][6:]

model = RandomForestClassifier(
    n_estimators=100,random_state=42,
)
model.fit(x_train,y_train)
prediction = model.predict(x_test)
print(prediction)
accuracy = accuracy_score(y_test, prediction)
print(accuracy)