Granite Timeseries Patchtst

🚀 Transformers.js

Transformers.jsは、ONNXウェイトを使用して、時系列予測を行うためのJavaScriptライブラリです。このライブラリは、ibm-granite/granite-timeseries-patchtstモデルを使用しており、Transformers.jsと互換性があります。

🚀 クイックスタート

インストール

まだの場合は、NPMからTransformers.js JavaScriptライブラリをインストールできます。

npm i @huggingface/transformers

使用例

基本的な使用法

// 時系列予測の例 w/ `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というサブフォルダに配置）。