neuralforecast包有很多引入好的时间序列预测算法模型,可以直接通过接口调用。
支持的算法模型有:
__all__ = ['RNN', 'GRU', 'LSTM', 'TCN', 'DeepAR', 'DilatedRNN',
'MLP', 'NHITS', 'NBEATS', 'NBEATSx', 'DLinear', 'NLinear',
'TFT', 'VanillaTransformer', 'Informer', 'Autoformer', 'PatchTST', 'FEDformer',
'StemGNN', 'HINT', 'TimesNet', 'TimeLLM', 'TSMixer', 'TSMixerx', 'MLPMultivariate',
'iTransformer', 'BiTCN', 'TiDE', 'DeepNPTS', 'SOFTS',