riba2534
79ff6dcccb
- 删除无用文件: PYEOF, PLAN.md, HURST_ENHANCEMENT_SUMMARY.md - 移动 REPORT.md → docs/REPORT.md，更新 53 处图片路径 - 移动 test_hurst_15scales.py → tests/，修复路径引用 - 清理 output/ 中未被报告引用的 60 个文件 - 重写 README.md 为开源标准格式（Badge、结构树、模块表等） - 添加 MIT LICENSE - 更新 .gitignore 排除运行时生成文件 Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>
76 lines
2.4 KiB
Python
76 lines
2.4 KiB
Python
#!/usr/bin/env python3
|
||
"""
|
||
测试脚本：验证Hurst分析增强功能
|
||
- 15个时间粒度的多尺度分析
|
||
- Hurst vs log(Δt) 标度关系图
|
||
"""
|
||
|
||
import sys
|
||
from pathlib import Path
|
||
|
||
# 添加项目路径
|
||
sys.path.insert(0, str(Path(__file__).parent))
|
||
|
||
from src.hurst_analysis import multi_timeframe_hurst, plot_multi_timeframe, plot_hurst_vs_scale
|
||
|
||
def test_15_scales():
|
||
"""测试15个时间尺度的Hurst分析"""
|
||
print("=" * 70)
|
||
print("测试15个时间尺度Hurst分析")
|
||
print("=" * 70)
|
||
|
||
# 定义全部15个粒度
|
||
ALL_INTERVALS = ['1m', '3m', '5m', '15m', '30m', '1h', '2h', '4h', '6h', '8h', '12h', '1d', '3d', '1w', '1mo']
|
||
|
||
print(f"\n将测试以下 {len(ALL_INTERVALS)} 个时间粒度：")
|
||
print(f" {', '.join(ALL_INTERVALS)}")
|
||
|
||
# 执行多时间框架分析
|
||
print("\n开始计算Hurst指数...")
|
||
mt_results = multi_timeframe_hurst(ALL_INTERVALS)
|
||
|
||
# 输出结果统计
|
||
print("\n" + "=" * 70)
|
||
print(f"分析完成：成功分析 {len(mt_results)}/{len(ALL_INTERVALS)} 个粒度")
|
||
print("=" * 70)
|
||
|
||
if mt_results:
|
||
print("\n各粒度Hurst指数汇总：")
|
||
print("-" * 70)
|
||
for interval, data in mt_results.items():
|
||
print(f" {interval:5s} | R/S: {data['R/S Hurst']:.4f} | DFA: {data['DFA Hurst']:.4f} | "
|
||
f"平均: {data['平均Hurst']:.4f} | 数据量: {data['数据量']:>7}")
|
||
|
||
# 生成可视化
|
||
output_dir = Path("output/hurst_test")
|
||
output_dir.mkdir(parents=True, exist_ok=True)
|
||
|
||
print("\n" + "=" * 70)
|
||
print("生成可视化图表...")
|
||
print("=" * 70)
|
||
|
||
# 1. 多时间框架对比图
|
||
plot_multi_timeframe(mt_results, output_dir, "test_15scales_comparison.png")
|
||
|
||
# 2. Hurst vs 时间尺度标度关系图
|
||
plot_hurst_vs_scale(mt_results, output_dir, "test_hurst_vs_scale.png")
|
||
|
||
print(f"\n图表已保存至: {output_dir.resolve()}")
|
||
print(" - test_15scales_comparison.png (15尺度对比柱状图)")
|
||
print(" - test_hurst_vs_scale.png (标度关系图)")
|
||
else:
|
||
print("\n⚠ 警告：没有成功分析任何粒度")
|
||
|
||
print("\n" + "=" * 70)
|
||
print("测试完成")
|
||
print("=" * 70)
|
||
|
||
if __name__ == "__main__":
|
||
try:
|
||
test_15_scales()
|
||
except Exception as e:
|
||
print(f"\n❌ 测试失败: {e}")
|
||
import traceback
|
||
traceback.print_exc()
|
||
sys.exit(1)
|