您当前所在位置:
首页 方法技巧 给一个完整的决策树预测3D和值大小的源代码,并加入决策树规则

给一个完整的决策树预测3D和值大小的源代码,并加入决策树规则

杨哥哥儿 177 2025-06-18 21:11
本帖最后由 杨哥哥儿 于 2025-6-18 21:26 编辑

<



该代码实现了完整的股票预测流程,包含数据获取、特征计算和决策树建模。核心决策规则通过export_text函数可视化输出,特征重要性通过柱状图展示。

典型决策规则可能包含"RSI<30且跨度增加则看涨"等逻辑。

代码说明:

使用pd.read_csv(r'3d_asc.txt')数据,计算5/10日均线、RSI等技术指标作为特征,决策树通过基尼系数选择最优分割点,最终输出可解释的和值大小的开出规则。

你就可以根据AI算出的规则去预测下和值的开出情况了。

规则结果如下:
<

算法是借股票的算法来完成的,可以较好的帮助大家理解。



更新了跨度计算公式,=|百-十|+|百-个|+|十-个|,没有了复杂的判断语句,值是跨度的2倍,不影响使用。