hex.tree.xgboost.task.XGBoostCleanupTask Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of h2o-ext-xgboost Show documentation

H2O XGBoost support

There is a newer version: 3.46.0.6

package hex.tree.xgboost.task;

import hex.tree.xgboost.matrix.RemoteMatrixLoader;

/**
 * Cleans up after XGBoost training
 */
public class XGBoostCleanupTask extends AbstractXGBoostTask {

  private XGBoostCleanupTask(XGBoostSetupTask setupTask) {
    super(setupTask);
  }

  @Override
  protected void execute() {
    XGBoostUpdater.terminate(_modelKey);
    RemoteMatrixLoader.cleanup(_modelKey.toString());
  }

  public static void cleanUp(XGBoostSetupTask setupTask) {
    new XGBoostCleanupTask(setupTask).doAllNodes();
  }

}