main.py 1.0 KB

1234567891011121314151617181920212223242526272829303132333435363738
  1. import get_data
  2. import get_data_by_sql
  3. import get_data_from_other_algorithm
  4. import data_process
  5. import core_algorithm
  6. # 数据库连接参数
  7. conn = connect(host, ......)
  8. # 准备算法输入参数
  9. parameter1 = *
  10. parameter2 = *
  11. # 多次调用核心算法时,将循环写在外面
  12. for (i=1:n){
  13. # 获取数据
  14. data1 = get_data(sn[i], start_time, end_time, ....) # 函数取数
  15. data2 = get_data_by_sql(sn[i], start_time, end_time, ....) # sql 语句直接数据库取数
  16. data3 = get_data_from_other_algorithm(data1, parameter1, ....) # 调用其他人的算法得到数据
  17. # 通用数据预处理 (可选,由算法说明文档说明算法输入数据是否需要预处理)
  18. data1 = data_process(data1)
  19. data2 = data_process(data2)
  20. # 调用核心算法
  21. [res1, res2] = core_algorithm(data1, data2, data3, parameter1, parameter2, ....)
  22. # 使用结果
  23. res1 = res1.append(res)
  24. res1.to_csv(...)
  25. res1.to_sql(...)
  26. }
  27. # 批量使用结果
  28. res.to_csv(...)
  29. res1.to_sql(...)