def data_to_label(strategy): print('getting new data to label') sample = Sample(train_obj.clf, data) if strategy == "random": print('random') X_TOLB, X_NOLB = sample.random(10) elif strategy == "entropy": print('entropy') X_TOLB, X_NOLB = sample.entropy(10) elif strategy == "entropydrop": print('entropy dropout') X_TOLB, X_NOLB = sample.entropy_dropout(10, 5) return (X_TOLB, X_NOLB)