candle.uq_utils.generate_index_distribution_from_block_list

candle.uq_utils.generate_index_distribution_from_block_list#

candle.uq_utils.generate_index_distribution_from_block_list(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 list of 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 (dictionary with parameters) – Contains the keywords that control the behavior of the function (uq_train_vec, uq_valid_vec, uq_test_vec)

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)