de.edux.data.provider.Dataset Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of lib Show documentation

Educational library for machine learning challenges

The newest version!

package de.edux.data.provider;

import java.util.List;
import java.util.Map;

public interface Dataset {

  double[][] getInputs(List dataset, int[] inputColumns);

  double[][] getTargets(List dataset, int targetColumn);

  Map getClassMap();

  double[][] getTrainFeatures(int[] inputColumns);

  double[][] getTrainLabels(int targetColumn);

  double[][] getTestLabels(int targetColumn);

  double[][] getTestFeatures(int[] inputColumns);

  String[] getColumnDataOf(int columnIndex);
}