====== AutoGluon ====== ^ file ^ class ^ link ^ | ''autogluon/timeseries/predictor.py'' | ''TimeSeriesPredictor'' | [[https://github.com/autogluon/autogluon/blob/1.2.0/timeseries/src/autogluon/timeseries/predictor.py#L47-L1570|link]] | [[https://wiki.yuuho.pro/doku.php?id=private:move:20241001|wiki > DLinear pytorch-lightning ミニバッチ生成]] ===== 時間計測関係 ===== * ''.fit()'' 学習・検証時間 * 各モデルごと * 検証ウィンドウ0 * モデル学習 * 検証データ内予測 & 評価 * 検証ウィンドウ1 * モデル学習 * 検証データ内予測 & 評価 * ensemble モデル * ''.predict()'' 予測時間 * 評価データ抽出 * fit * 統計系はダミーを返している? * ディープ系は学習が走っている * val * 統計系はここでパラメータ選定と予測と評価をしている * ディープ系も推論と評価 * predict predictor.fit learner.fit learner._fit trainer.fit <- AutoTimeSeriesTrainer (not AbstractTimeSeriesTrainer) trainer._train_multi for i, model in enumerate(models): ハイパラチューニングモデル or トレーニングモデル trainer.tune_model_hyperparameters trainer._train_and_save trainer._train_single model.fit model._fit for window_index, (train_fold, val_fold) in enumerate(): model.fit model.score_and_cache_oof trainer._log_scores_and_times trainer.save_model predictor.refit_full predictor.predict learner.predict trainer.predict trainer.get_model_pred_dict | モデル準備 | for model_name in model_set: | if model_name not in model_pred_dict: | model_pred_dict[model_name] = trainer._predict_model() model.predict() * multiwindow の fit では各ウィンドウごとに毎回 refit (model.fit()が走る)