前言
本文基于开源AI交互式框架tymon,对航班人数进行预测分析。
实战
安装tymon
pip install tymon
或者git clone 代码仓
git clone https://github.com/TymonXie/tymon.git
推荐使用pip,会将以来的pytorch等进行安装,数据可在example文件夹下获得
使用tymon
从tymon中导入时序助手 TimeSeries,并实例化,输入想要使用的模型和所用数据地址,并调用run方法。
assistant_object = TimeSeries(model_name='LSTM', data_path='international-airline-passengers.csv') assistant_object.run()
会自动弹出参数设置窗口,如下:
有默认参数,你也可以根据需要修改相应参数,并点击start,开始训练模型
训练完成后,会在当前目录得到模型,以及最终的性能展示。
至此,一次基于LSTM预测航班人数的实战就完成了,框架很多能力还处于初期,比如数据格式的支持,还有参数搜索等,当然这些都是使用者不感知的,下次更新了再写,respect!
如果你觉得有用或者想参与到该框架中,项目地址如下,点个star哇!