- log_string(f"generating {nb} samples (can take some time)")
- self.train_descr = generate_descr(
- (nb * 4) // 5, "train", pruner=self.pruner_train
- )
- self.test_descr = generate_descr((nb * 1) // 5, "test", pruner=None)
+ log_string(f"generating {nb_train_samples+nb_test_samples} samples (can take some time)")
+ self.train_descr = generate_descr(nb_train_samples, "train", pruner=self.pruner_train)
+ self.test_descr = generate_descr(nb_test_samples, "test", pruner=None)