python
import pandas as pd
from sklearn.model_selection import train_test_split
from sklearn.ensemble import RandomForestRegressor
from sklearn.metrics import mean_squared_error
加载数据
df = pd.read_csv('stock_data.csv')
准备特征(X)和目标(y)
X = df.drop(['收盘价'], axis=1) 假设我们预测收盘价
y = df['收盘价']
训练集和测试集
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2)
初始化并训练模型
model = RandomForestRegressor()
model.fit(X_train, y_train)