com.expleague.ml.cli.builders.data.impl.DataBuilderClassic Maven / Gradle / Ivy

Go to download
package com.expleague.ml.cli.builders.data.impl;

import com.expleague.commons.util.Pair;
import com.expleague.ml.cli.builders.data.DataBuilder;
import com.expleague.ml.cli.builders.data.PoolReader;
import com.expleague.ml.data.tools.Pool;

/**
 * User: qdeee
 * Date: 03.09.14
 */
public class DataBuilderClassic implements DataBuilder {
  private String learnPath;
  private String testPath;
  private PoolReader reader;

  @Override
  public void setLearnPath(final String learnPath) {
    this.learnPath = learnPath;
  }

  @Override
  public void setReader(final PoolReader reader) {
    this.reader = reader;
  }

  public void setTestPath(final String testPath) {
    this.testPath = testPath;
  }


  @Override
  public Pair create() {
    final Pool learn = reader.read(learnPath);
    final Pool test;
    if (testPath != null && !testPath.equals(learnPath)) {
      test = reader.read(testPath);
    }
    else {
      test = learn;
    }
    return Pair.create(learn, test);
  }
}