Class Model

core/src/main/java/io/ddf/ml/Model.java:20–220

 */

public class Model implements IModel, Serializable {

  private static final long serialVersionUID = 8076703024981092021L;

  private Object mRawModel;

  private String modelType;

  private String mName;

  private String[] mTrainedColumns;

  private String mClass = this.getClass().getName(); //for serialization

  public Model(Object rawModel) {
    mRawModel = rawModel;

    if (rawModel != null) {
      modelType = mRawModel.getClass().getName();
    } else {
      modelType = null;
    }

    mName = UUID.randomUUID().toString();
  }

  @Override
  public Object getRawModel() {
    return mRawModel;
  }

  @Override
  public void setRawModel(Object rawModel) {
    this.mRawModel = rawModel;
    this.modelType = rawModel.getClass().getName();
  }


  @Override
  public String getName() {
    return mName;
  }

  @Override
  public void setName(String name) {
    mName = name;
  }

  @Override
  public void setTrainedColumns(String[] columns) {
    this.mTrainedColumns = columns;
  }

  @Override
  public String[] getTrainedColumns() {
    return this.mTrainedColumns;
  }