hookeai.miscellaneous.pruning.pruning_dataset.read_samples_loss_from_dir

read_samples_loss_from_dir(predictions_dir)

Read loss samples from prediction directory.

Parameters:

predictions_dir (str) – Directory where samples predictions results files are stored.

Returns:

• samples_id (list[int]) – Samples IDs.

• samples_loss (list[float]) – Samples prediction loss.