)
# 特征工程
# 计算一些技术指标作为特征
data['SMA_20'] = data['Close'].rolling(window=20).mean() # 20日简单移动平均
data['RSI'] = 100 - (100 / (1 + (data['Close'].pct_change().rolling(window=14).mean() / data['Close'].pct_change().rolling(window=14).std())))
data['Return'] = data['Close'].pct_change() # 日收益率
data['Volatility'] = data['Return'].rolling(window=5).std() * np.sqrt(252) # 5日波动率，年化
# 创建目标变量，假设我们要预测下一天的收益率
data['Next_Day_Return'] = data['Return'].shift(-1)
# 去掉NaN值
data = data.dropna()
# 准备特征和目标变量
X = data[['Open', 'High', 'Low', 'Close', 'Volume', 'SMA_20', 'RSI', 'Volatility']]
y = data['Next_Day_Return']
# 数据分割
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2, random_state=42)
# 数据标准化
scaler = StandardScaler()
X_train_scaled = scaler.fit_transform(X_train)
X_test_scaled = scaler.transform(X_test)
# 训练随机森林模型
rf_model = RandomForestRegressor(n_estimators=100, random_state=42)
rf_model.fit(X_train_scaled, y_train)
# 预测
predictions = rf_model.predict(X_test_scaled)
# 评估模型
mse = mean_squared_error(y_test, predictions)
r2 = r2_score(y_test, predictions)
print(f"均方误差: {mse}")
print(f"R² 得分: {r2}")
# 特征重要性
feature_importance = pd.DataFrame({'feature': X.columns, 'importance': rf_model.feature_importances_})
print("特征重要性:")
print(feature_importance.sort_values('importance', ascending=False))
# 预测新数据
# 假设我们有新的股票数据要预测
new_data = pd.DataFrame({
'Open': [100.5],
'High': [101.2],
'Low': [99.8],
'Close': [100.1],
'Volume': [500000],
'SMA_20': [100.0],
'RSI': [50.0],
'Volatility': [0.2]
})
# 标准化新数据
new_data_scaled = scaler.transform(new_data)
# 预测
predicted_return = rf_model.predict(new_data_scaled)
print("预测的下一日收益率:", predicted_return[0]) 加载方法： 将上述代码保存为一个Python文件，例如"AI_Stock_Picking.py"。然后使用Python环境运行此脚本，确保安装了所需的库（pandas, numpy, sklearn）。你可以通过命令行运行： python AI_Stock_Picking.py 参数说明 参数 意义 n_estimators 随机森林中树的数量，影响模型的复杂度和精度 random_state 随机种子，用于保证结果的可复现性 test_size 测试集数据比例，常用于验证模型性能 feature_importances_ 展示每个特征对预测结果的重要性 使用建议 此AI选股策略适用于较长期的投资分析。在使用时，建议： 使用更多的特征，包括基本面数据（如市盈率、市净率、财报数据等），以提高模型的预测能力。 定期更新模型，根据市场变化调整特征和参数。 结合多种模型（如SVM、神经网络等）进行集成学习，可能提高预测的准确性。 注意模型的过拟合问题，特别是在数据量较少的情况下，可以通过交叉验证来优化。 设置合理的预期和止损点，机器学习预测并非完美，需结合市场实际情况。 X用户点评 "AI选股让我找到了一些有潜力的股票，但记得要结合市场分析，不能全靠模型。" - @AIInvestor "在股票市场用机器学习选股时，要注意特征的选择和模型的更新，市场总是变化的。" - @DataDrivenTrader "这个策略在期货市场也能用，但要特别注意数据的实时性和模型的快速响应。" - @FuturesQuant "外汇市场的AI选股需要考虑到全球经济事件的影响，单纯的技术指标可能不够。" - @ForexML "加密货币市场的波动性让这个策略需要谨慎对待，模型可能很快失效。" - @CryptoAI 来源：今日美股网lg...
北美站
