オープンソースの AutoML および PyTorch モデル トレーニング ライブラリ
ドキュメント |例
!!!注意 GradsFlow は急速に変化しています。 0.1.0
に到達するまでは、多くの重大な変更が行われる予定です。問題を作成してフィードバックをお寄せいただくか、Slack グループに参加してください。
GradsFlow は、PyTorch に基づくオープンソースの AutoML ライブラリです。私たちの目標は、AI を民主化し、誰もが利用できるようにすることです。
さまざまなタスクのディープ ラーニング モデルをラップトップ上で、またはラップトップから直接リモート クラスターに自動的に構築およびトレーニングできます。ほぼすべての PyTorch モデルのトレーニングに使用できる、強力で拡張が簡単なモデル トレーニング API を提供します。 GradsFlow には独自のモデル トレーニング API がありますが、PyTorch Lightning Flash もサポートしており、さまざまなタスクにわたってより豊富な機能を提供します。
!!! info Gradsflow は初心者と専門家の両方のために構築されています。 AutoTasks
ゼロコード AutoML を提供し、 Model
and Tuner
カスタム モデル トレーニングとハイパーパラメーターの最適化を提供します。
推奨:
gradsflow
をインストールする推奨方法は、PyPI からのpip
を使用するか、conda-forge チャネルからのconda
を使用することです。
ピップ付き
pip install -U gradsflow
コンダ付き
conda install -c conda-forge gradsflow
最新(不安定):
十分なやる気があれば、次のようにgradsflow
の最新の最先端バージョン (不安定な可能性があります) をインストールすることもできます。
pip install git+https://github.com/gradsflow/gradsflow@main
あなたは初心者ですか、それとも機械学習以外のバックグラウンドを持っていますか?このセクションはあなたのためのものです。 Gradsflow AutoTask
画像認識、感情分析、テキスト要約などを含むさまざまなタスクにわたる自動モデル構築とトレーニングを提供します。
Tuner
、モデル トレーニングからハイパーパラメーターの最適化に移行するための簡素化された API を提供します。
gradsflow.core
: コアは AutoML タスクの構成要素を定義します。
gradsflow.autotasks
: AutoTasks は、gradsflow AutoML API によって提供されるさまざまな ML/DL タスクを定義します。
gradsflow.model
: GradsFlow モデルは、シンプルでありながらカスタマイズ可能なモデル トレーニング API を提供します。 model.fit(...)
を使用して任意の PyTorch モデルをトレーニングでき、より複雑なタスクに合わせて簡単にカスタマイズできます。
gradsflow.tuner
: コード変更を最小限に抑えた AutoModel HyperParameter 検索。
?詳細については、ノートブックの例を確認してください。
? ko-fiのスポンサー
?サポートが必要ですか? [email protected] までお問い合わせください
ソーシャル: Twitter @gradsflow や Linkedin でフォローして最新情報を入手することもできます。
Slack グループに参加してチャットしてください。
あらゆる種類の貢献を歓迎します。ドキュメントの更新、例の追加、特定された問題の修正、新機能の追加/リクエストを行うことができます。
詳細については、投稿する前に投稿ガイドラインを確認してください。
私たちは、オープンで、歓迎的で、多様性があり、包括的で健全なコミュニティに貢献する方法で行動し、交流することを誓います。
貢献者規約の行動規範を全文読む
GradsFlowは、素晴らしいオープンソース プロジェクト (Ray、PyTorch Lightning、HuggingFace Accelerate、TorchMetrics を含むがこれらに限定されない) の助けを借りて構築されています。複数のプロジェクト Keras と FastAI からインスピレーションを得ています。