diff --git a/scripts/ml/ml_functions/test_func_singlegene.py b/scripts/ml/ml_functions/test_func_singlegene.py index 2b1f3ac..86f212d 100644 --- a/scripts/ml/ml_functions/test_func_singlegene.py +++ b/scripts/ml/ml_functions/test_func_singlegene.py @@ -130,3 +130,6 @@ df2 = split_tts(df all(df2['X'].columns.isin(['gene_name'])) # should be False +from sklearn.utils import all_estimators + +all_clfs = all_estimators(type_filter="classifier")