1. 学ぶ
    2. /
  2. コース
    3. /
  3. MLflow 入門
Connected

演習

カスタムPythonクラスの作成

MLflowは、多様なユースケースに対応できるよう、カスタムモデルを作成する方法を提供します。カスタムモデルを作成するには、mlflow.pyfunc.PythonModel クラスを継承するPythonクラスを作成します。PythonModel クラスは、カスタムの推論ロジックやアーティファクト依存関係のためのメソッドを提供し、柔軟に拡張できます。

この演習では、特定のモデルを読み込み、推論後にラベルをデコードするカスタムモデル用の新しいPythonクラスを作成します。mlflow モジュールはすでにインポートされます。

指示

100 XP
  • CustomPredict という名前のPythonクラスを作成します。
  • カスタムクラス内でアーティファクトを読み込むために使う load_context() メソッドを定義します。
  • カスタム推論を定義するための predict() メソッドを定義します。