Hub documentation

Hugging Faceでのfastaiの活用

Hugging Face's logo
Join the Hugging Face community

and get access to the augmented documentation experience

to get started

Hugging Faceでのfastaiの活用

fastai は、PyTorch と Python を活用したオープンソースの Deep Learning ライブラリです。テキスト、ビジョン、テーブルデータに対して、最先端の出力を得ながら、高速かつ高精度なニューラルネットワークを訓練するための高レベルなコンポーネントを提供します。

Hubでのfastaiの探索

モデルページの左側でフィルタリングすることで、fastaiのモデルを見つけることができます。

Hub上のすべてのモデルには、以下の機能が備わっています:

  1. 検索性を高めるための簡潔な説明とメタデータタグを備えた、自動生成されたモデルカード。
  2. ブラウザ上でモデルを直接試すことができるインタラクティブなウィジェット(Image Classification用)
  3. Image Classification用の推論リクエストを送信できるInference Providersウィジェットです。

既存のモデルを使用する

huggingface_hub ライブラリは、Hub からモデルをダウンロードするためのユーティリティ関数を備えた軽量な Python クライアントです。

pip install huggingface_hub["fastai"]

ライブラリをインストールした後、from_pretrained_fastai メソッドを使用するだけです。このメソッドはモデルを読み込むだけでなく、モデルが保存された時点のfastaiバージョンも検証します。再現性にとって重要な点です。

from huggingface_hub import from_pretrained_fastai

learner = from_pretrained_fastai("espejelomar/identify-my-cat")

_,_,probs = learner.predict(img)
print(f"Probability it's a cat: {100*probs[1].item():.2f}%")

# Probability it's a cat: 100.00%

特定のモデルの読み込み方法を確認するには、Use in fastai をクリックしてください。動作するスニペットが表示され、モデルを読み込むことができます。

モデルの共有

fastaiモデルは、push_to_hub_fastaiメソッドを使用して共有できます。

from huggingface_hub import push_to_hub_fastai

push_to_hub_fastai(learner=learn, repo_id="espejelomar/identify-my-cat")

追加のリソース

Update on GitHub