python
import pandas as pd
from sklearn.ensemble import RandomForestClassifier
from sklearn.model_selection import train_test_split
from sklearn.metrics import accuracy_score
加载数据
data = pd.read_csv('stock_data.csv') 假设数据包含特征列和标签列
特征和标签
X = data.drop(columns=['Label']) 特征
y = data['Label'] 标签(例如1表示买入,0表示不买)
划分训练集和测试集
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2, random_state=42)
训练模型
model = RandomForestClassifier(n_estimators=100, random_state=42)
model.fit(X_train, y_train)