Skip to content

Instantly share code, notes, and snippets.

@agibsonccc
Created August 27, 2014 02:59
Show Gist options
  • Save agibsonccc/b11dbf9adbb44350644d to your computer and use it in GitHub Desktop.
Save agibsonccc/b11dbf9adbb44350644d to your computer and use it in GitHub Desktop.
package org.deeplearning4j.linalg.dataset.api;
import org.apache.commons.math3.random.RandomGenerator;
import org.deeplearning4j.linalg.api.ndarray.INDArray;
import org.deeplearning4j.linalg.dataset.SplitTestAndTrain;
import java.io.Serializable;
import java.util.Iterator;
import java.util.List;
/**
* Created by agibsonccc on 8/26/14.
*/
public interface DataSet extends Iterable<org.deeplearning4j.linalg.dataset.DataSet>,Serializable {
INDArray getFeatures();
void setFeatures(INDArray features);
void setLabels(INDArray labels);
org.deeplearning4j.linalg.dataset.DataSet copy();
org.deeplearning4j.linalg.dataset.DataSet reshape(int rows, int cols);
void multiplyBy(double num);
void divideBy(int num);
void shuffle();
void squishToRange(double min, double max);
void scale();
void addFeatureVector(INDArray toAdd);
void addFeatureVector(INDArray feature, int example);
void normalize();
void binarize();
void binarize(double cutoff);
void normalizeZeroMeanZeroUnitVariance();
int numInputs();
void validate();
int outcome();
void setNewNumberOfLabels(int labels);
void setOutcome(int example, int label);
org.deeplearning4j.linalg.dataset.DataSet get(int i);
org.deeplearning4j.linalg.dataset.DataSet get(int[] i);
List<List<org.deeplearning4j.linalg.dataset.DataSet>> batchBy(int num);
org.deeplearning4j.linalg.dataset.DataSet filterBy(int[] labels);
void filterAndStrip(int[] labels);
List<org.deeplearning4j.linalg.dataset.DataSet> dataSetBatches(int num);
List<List<org.deeplearning4j.linalg.dataset.DataSet>> sortAndBatchByNumLabels();
List<List<org.deeplearning4j.linalg.dataset.DataSet>> batchByNumLabels();
List<org.deeplearning4j.linalg.dataset.DataSet> asList();
SplitTestAndTrain splitTestAndTrain(int numHoldout);
INDArray getLabels();
INDArray getFeatureMatrix();
void sortByLabel();
void addRow(org.deeplearning4j.linalg.dataset.DataSet d, int i);
INDArray exampleSums();
INDArray exampleMaxs();
INDArray exampleMeans();
org.deeplearning4j.linalg.dataset.DataSet sample(int numSamples);
org.deeplearning4j.linalg.dataset.DataSet sample(int numSamples, RandomGenerator rng);
org.deeplearning4j.linalg.dataset.DataSet sample(int numSamples, boolean withReplacement);
org.deeplearning4j.linalg.dataset.DataSet sample(int numSamples, RandomGenerator rng, boolean withReplacement);
void roundToTheNearest(int roundTo);
int numOutcomes();
int numExamples();
List<String> getLabelNames();
void setLabelNames(List<String> labelNames);
List<String> getColumnNames();
void setColumnNames(List<String> columnNames);
@Override
Iterator<org.deeplearning4j.linalg.dataset.DataSet> iterator();
}
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment