Granite Timeseries Patchtst

🚀 transformers.js

transformers.js 是一個與 ibm-granite/granite-timeseries-patchtst 基礎模型適配的庫，可用於時間序列預測任務。它藉助 ONNX 權重，實現了與 Transformers.js 的兼容，為時間序列預測提供了便捷的解決方案。

🚀 快速開始

安裝

若你還未安裝 Transformers.js JavaScript 庫，可使用以下命令從 NPM 進行安裝：

npm i @huggingface/transformers

💻 使用示例

基礎用法

以下是使用 onnx-community/granite-timeseries-patchtst 進行時間序列預測的示例代碼：

import { PatchTSTForPrediction, Tensor } from "@huggingface/transformers";

const model_id = "onnx-community/granite-timeseries-patchtst";
const model = await PatchTSTForPrediction.from_pretrained(model_id, { dtype: "fp32" });

const dims = [64, 512, 7];
const prod = dims.reduce((a, b) => a * b, 1);
const past_values = new Tensor('float32',
    Float32Array.from({ length: prod }, (_, i) => i / prod),
    dims,
);
const { prediction_outputs } = await model({ past_values });
console.log(prediction_outputs);

📚 詳細文檔

注意事項

目前為 ONNX 權重單獨創建倉庫只是一個臨時解決方案，直到 WebML 獲得更廣泛的應用。若你想讓自己的模型適用於 Web，建議使用 🤗 Optimum 將模型轉換為 ONNX 格式，並按照本倉庫的結構進行組織（將 ONNX 權重存放在名為 onnx 的子文件夾中）。