candle.uq_utils.generate_index_distribution_from_blocks

candle.uq_utils.generate_index_distribution_from_blocks#

candle.uq_utils.generate_index_distribution_from_blocks(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 block quantities and that the block quantities 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 (dictionary with parameters) – Contains the keywords that control the behavior of the function (uq_train_bks, uq_valid_bks, uq_test_bks)

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)