pytorch-forecasting

收录理由

这个库把Temporal Fusion Transformer、DeepAR、N-BEATS和N-HiTS等一系列深度学习预测模型，统一封装在基于PyTorch Lightning的高层API之下。TimeSeriesDataSet类把预测数据里那些容易出错的部分都处理好了，比如分组ID、静态和时变协变量、编码器和解码器长度，以及训练验证集的划分，这样从原始表格到训练好的模型，不用自己费力搭数据管道。它采用分位数损失，能给出真正的不确定性区间，而不只是点预测，这对做规划和库存决策来说尤其实用。由于这些模型直接接入Lightning的训练器，GPU扩展、检查点保存、早停机制，还有配合Optuna做超参数调优，几乎都是开箱即用。对于想要用一套代码同时应付实验和生产环境预测、不想东拼西凑写临时脚本的团队，这是个很合适的基础。