panda/bettafish-company
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

预测函数调通

Browse Source
This commit is contained in:
juanboy
2024-07-04 18:07:35 +08:00
parent f7939fa6c7
commit 59b18fa5a2
1 changed files with 0 additions and 36 deletions
Download Patch File Download Diff File Expand all files Collapse all files
predict_demo.py
-36
View File
@@ -1,36 +0,0 @@
'''
用于测试预测逻辑
@Author: QST520
'''
import numpy as np
import datetime
import matplotlib.pyplot as plt
def datetime_to_number(date: str): # 格式化日期转换为 integer
date_number = datetime.datetime.strptime(date, "%Y-%m-%d")
base_number = datetime.datetime.strptime("2024-1-1", "%Y-%m-%d")
return date_number.__sub__(base_number).days
if __name__ == '__main__': # 预测 demo
xs = [
'2024-6-4', '2024-6-5', '2024-6-6', '2024-6-7', '2024-6-8', '2024-6-9',
'2024-6-10', '2024-6-11', '2024-6-12', '2024-6-13'
]
ys = [15, 14, 16, 15, 16, 13, 12, 11, 9, 8]
xs = np.array(list(map(datetime_to_number, xs)))
ys = np.array(ys)
fit = np.polyfit(xs, ys, 1)
fn = np.poly1d(fit)
print('2024-6-14 PREDICTION: ' +
str(int(fn(datetime_to_number('2024-6-14')))))
print('2024-6-15 PREDICTION: ' +
str(int(fn(datetime_to_number('2024-6-15')))))
print('2024-6-16 PREDICTION: ' +
str(int(fn(datetime_to_number('2024-6-16')))))
print('2024-6-17 PREDICTION: ' +
str(int(fn(datetime_to_number('2024-6-17')))))
print('2024-6-18 PREDICTION: ' +
str(int(fn(datetime_to_number('2024-6-18')))))