一种简便的时间序列预测深度模型
deepforecast的Python项目详细描述
DeepForecast
一种易于使用的时间序列预测深度模型
方法
- STTF
- 序号2
如何安装
pip install deepforecast
如何使用
以STTF为例:
fromtensorflow.keras.utilsimportplot_modelfromdeepforecast.featuresimportSparseColumn,SequenceColumnfromdeepforecast.modelsimportSTTFattr_feats=["age","user","platform"]sequence_feats=["history","future"]attr_columns=[]forfeatinattr_feats:col=SparseColumn(name=feat,vocab_size=10,embed_dim=8)attr_columns.append(col)sequence_columns=[]hist_col=SequenceColumn(name="history",num_seq=5,seq_steps=28,dim=1)sequence_columns.append(hist_col)fut_col=SequenceColumn(name="future",num_seq=4,seq_steps=7,dim=1)sequence_columns.append(fut_col)model=STTF(attr_columns,sequence_columns,attr_attention_embed_dim=12)model.summary()plot_model(model,show_shapes=True)model.compile(optimizer="rmsprop",loss=["mse","mse"],loss_weights=[0.2,0.8],metrics=["mse"])
- 项目
标签: