Sibyl: 时序预测包

Sibyl是一个使用PyTorchLightning进行时序数据预测的包，PyTorchLightning是一个为机器学习研究人员提供的轻量级PyTorch包装器。

Sibyl由Microsoft Substrate Connectivity Team发布和维护。

模型

• 序列到序列
• 长短时记忆
• 神经网络
• 注意力神经网络

安装

Sibyl可在PyPi上找到，您可以通过pip免费获取。

pip install mssibyl

快速入门

更新中...

import pandas as pd
from mssibyl.models import BaseModel
from mssibyl.trainer import BaseTrainer
from mssibyl.datasets.utils import make_timeline

df = pd.read_csv("./data.csv")
df.columns = ["ds", "y"]

df_with_future = make_timeline(df, freq="H", periods=300)

args = {
    "lookback": 48,
    "forecast_horizon": 1,
    "data": df_with_future,
    "target": "y",
}

sibyl_model = BaseModel(args)
trainer = BaseTrainer(
    max_epochs=100, gpus=1, show_progress_bar=False, early_stop_callback=True
)
trainer.fit(sibyl_model)
preds = sibyl_model.predict(interval=True, include_history=True)

preds.to_csv("res.csv", index=False)

变更日志

版本 0.1 (即将推出)

• 初始发布

许可证

Sibyl遵循MIT许可证