com.logicalclocks.hsfs.spark

Class TrainingDatasetBundle

java.lang.Object

com.logicalclocks.hsfs.spark.TrainingDatasetBundle

public class TrainingDatasetBundle
extends Object

Constructor Summary

Constructors

Constructor and Description
TrainingDatasetBundle(Integer version)
TrainingDatasetBundle(Integer version, org.apache.spark.sql.Dataset<org.apache.spark.sql.Row> dataset, List<String> labels)
TrainingDatasetBundle(Integer version, Map<String,org.apache.spark.sql.Dataset<org.apache.spark.sql.Row>> datasetSplits, List<String> labels)

Method Summary

Modifier and Type	Method and Description
List<org.apache.spark.sql.Dataset<org.apache.spark.sql.Row>>	getDataset(Boolean splitLabels)
List<org.apache.spark.sql.Dataset<org.apache.spark.sql.Row>>	getDataset(String split, Boolean splitLabels)

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Constructor Detail

TrainingDatasetBundle

public TrainingDatasetBundle(Integer version,
                             org.apache.spark.sql.Dataset<org.apache.spark.sql.Row> dataset,
                             List<String> labels)

TrainingDatasetBundle

public TrainingDatasetBundle(Integer version)

TrainingDatasetBundle

public TrainingDatasetBundle(Integer version,
                             Map<String,org.apache.spark.sql.Dataset<org.apache.spark.sql.Row>> datasetSplits,
                             List<String> labels)

Method Detail

getDataset

public List<org.apache.spark.sql.Dataset<org.apache.spark.sql.Row>> getDataset(Boolean splitLabels)

getDataset

public List<org.apache.spark.sql.Dataset<org.apache.spark.sql.Row>> getDataset(String split,
                                                                               Boolean splitLabels)