Class TrainingDatasetBundle

java.lang.Object
com.logicalclocks.hsfs.spark.TrainingDatasetBundle

public class TrainingDatasetBundle extends Object
  • Constructor Details

    • 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 Details

    • 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)