You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
22 lines
648 B
22 lines
648 B
from typing import List, Type
|
|
import pg_request as pg
|
|
import models as m
|
|
import echarts_visualization as ev
|
|
|
|
|
|
def main(target: str,
|
|
target_name: str,
|
|
exog_columns: List[str],
|
|
models: List[Type] = [
|
|
m.VAR_Forecasting, m.ARIMA_Forecasting, m.SARIMA_Forecasting,
|
|
m.RF_Forecasting
|
|
]):
|
|
pg.run()
|
|
m.run(forecast_target=target, exog_columns=exog_columns, models=models)
|
|
ev.run(target=target, target_name=target_name, models=models)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
main('liugan_index', '流感指数',
|
|
['infection_number.1', 'infection_number.2', 'jijin_data', 'shoupan'])
|