軸承是許多機械設備中的關鍵部件。透過監測軸承的運行數據,我們可以預測軸承的健康狀態,並在需要時進行維護,從而避免突發故障。
import pandas as pd
from sklearn.ensemble import RandomForestClassifier
from sklearn.metrics import accuracy_score, classification_report
from sklearn.model_selection import train_test_split
# 假設你已經有了一個包含特徵和目標變量的數據集 df
# 提取特徵和目標變量
X = df[['Vibration', 'Temperature', 'Speed']]
y = df['Status'] # Status: 0 表示正常,1 表示異常
# 將數據集劃分為訓練集和測試集
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2, random_state=42)
# 初始化隨機森林分類器
rf_model = RandomForestClassifier()
# 訓練模型
rf_model.fit(X_train, y_train)
# 預測軸承狀態
y_pred = rf_model.predict(X_test)
# 計算準確度
accuracy = accuracy_score(y_test, y_pred)
print(f"準確度: {accuracy}")
print(classification_report(y_test, y_pred))
這個示例假設你已經有了一個包含特徵(如振動、溫度、速度等)和目標變量(軸承狀態)的數據集~