Add documentation of '_translate_train_sizes'