ai.h2o.automl.H2OJob Maven / Gradle / Ivy

Go to download

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

H2O AutoML

The newest version!

package ai.h2o.automl;

import water.H2O;
import water.Job;
import water.Key;
import water.Keyed;


public class H2OJob {

  protected final T _target;
  protected Key _jobKey;
  Job _job;

  public H2OJob(T runnable, long max_runtime_msecs) {
    this(runnable, Key.make(), max_runtime_msecs);
  }

  public H2OJob(T runnable, Key key, long max_runtime_msecs) {
    _target = runnable;
    _job = new Job<>(key, _target.getClass().getName(), _target.getClass().getSimpleName() + " build");
    _jobKey = _job._key;
    _job._max_runtime_msecs = max_runtime_msecs;
  }

  public Job start() {
    return this.start(1);
  }

  public Job start(int work) {
    return _job.start(new H2O.H2OCountedCompleter() {
      @Override public void compute2() {
        _target.run();
        tryComplete();
      }
    }, work);
  }

  public void stop() { _jobKey.get().stop(); }
}