ncem.estimators.EstimatorCVAEncem.split_data_node

EstimatorCVAEncem.split_data_node(test_split: float, validation_split: float, seed: int = 1)

Split nodes randomly into partitions.

Parameters

• test_split (float) – Fraction of total nodes to be in test set.

• validation_split (float) – Fraction of train-eval nodes to be in validation split.

• seed (int) – Seed for random selection of observations.

Raises

ValueError – If evaluation or test dataset are empty.