candle.uq_utils.generate_index_distribution_from_fraction

candle.uq_utils.generate_index_distribution_from_fraction(numTrain, numTest, numValidation, params)

Generates a vector of indices to partition the data for training. It checks that the fractions provided are (0, 1) and add up to 1.

Parameters:

• numTrain (int) – Number of training data points

• numTest (int) – Number of testing data points

• numValidation (int) – Number of validation data points (may be zero)

• params (dictionary with parameters) – Contains the keywords that control the behavior of the function (uq_train_fr, uq_valid_fr, uq_test_fr)

Returns:

• indexTrain (int numpy array) – Indices for data in training

• indexValidation (int numpy array) – Indices for data in validation (if any)

• indexTest (int numpy array) – Indices for data in testing (if merging)