Ctrl+K

candle.uq_utils.generate_index_distribution

candle.uq_utils.generate_index_distribution#

candle.uq_utils.generate_index_distribution(numTrain, numTest, numValidation, params)#

Generates a vector of indices to partition the data for training. NO CHECKING IS DONE: it is assumed that the data could be partitioned in the specified blocks and that the block indices describe a coherent partition.

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 (Dict) – Contains the keywords that control the behavior of the function (uq_train_fr, uq_valid_fr, uq_test_fr for fraction specification, uq_train_vec, uq_valid_vec, uq_test_vec for block list specification, and uq_train_bks, uq_valid_bks, uq_test_bks for block number specification)

Returns:

Tuple of numpy arrays

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

Contents