保存はtxt形式?

LightGBMのsave_model関数によって保存されるモデルは、テキスト形式のモデルファイルですが、拡張子が.txtではなく、.jsonまたは.txtのいずれかです。LightGBMはモデルをJSON形式で保存しますが、一般的には.json拡張子が使用されます。ただし、.txt拡張子も使用できます。

以下は、LightGBMモデルを.jsonファイルとして保存する例です:

python
import lightgbm as lgb # モデルの訓練 lgbm_model = lgb.LGBMClassifier(n_estimators=100) lgbm_model.fit(X_train, y_train) # モデルの保存 lgbm_model.save_model('lgbm_model.json') # モデルの再読み込み loaded_lgbm_model = lgb.Booster(model_file='lgbm_model.json') # 新しいデータでの予測 predictions = loaded_lgbm_model.predict(X_test)

このようにして、LightGBMモデルをJSON形式のファイルとして保存し、後で再読み込みして使用することができます。

未分類

Posted by ぼっち