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

Go to download

Show more of this group Show more artifacts with this name
Show all versions of jmll

Various ML methods implemented by myself and my students

There is a newer version: 1.4.9

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);
  }
}