core/src/main/java/io/prediction/controller/java/WorkflowParamsBuilder.java

/**
 * Copyright 2015 TappingStone, Inc.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package io.prediction.controller.java;

import io.prediction.controller.WorkflowParams;

/**
 * Builder for WorkflowParams.
 */
public class WorkflowParamsBuilder {
  private String batch = "";
  private int verbose = 2;
  private boolean saveModel = false;

  /**
   * Initializes a builder with default values.
   */
  public WorkflowParamsBuilder() {}

  /**
   * Sets the batch label.
   */
  public WorkflowParamsBuilder batch(String batch) {
    this.batch = batch;
    return this;
  }

  /**
   * Sets the verbosity level.
   */
  public WorkflowParamsBuilder verbose(int verbose) {
    this.verbose = verbose;
    return this;
  }

  /**
   * Sets whether trained models are persisted.
   */
  public WorkflowParamsBuilder saveModel(boolean saveModel) {
    this.saveModel = saveModel;
    return this;
  }

  /**
   * Build a WorkflowParams object.
   */
  public WorkflowParams build() {
    return new WorkflowParams(batch, verbose, saveModel);
  }
}