Granite Timeseries Patchtsmixer

🚀 Transformers.js

Transformers.jsは、ONNXウェイトを使用して、https://huggingface.co/ibm-granite/granite-timeseries-patchtsmixer をWeb環境で利用可能にするライブラリです。時系列予測のタスクに最適化されています。

🚀 クイックスタート

📦 インストール

まだインストールしていない場合は、Transformers.js JavaScriptライブラリをNPMから以下のコマンドでインストールできます。

npm i @huggingface/transformers

💻 使用例

基本的な使用法

// このコードは、onnx-community/granite-timeseries-patchtsmixerを使った時系列予測の例です。
import { PatchTSMixerForPrediction, Tensor } from '@huggingface/transformers';

const model_id = "onnx-community/granite-timeseries-patchtsmixer";
const model = await PatchTSMixerForPrediction.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というサブフォルダに配置します）。

情報一覧