added model names sklearn

This commit is contained in:
Tanushree Tunstall 2022-07-16 15:35:42 +01:00
parent 39e72b2dfb
commit 0adf69f75a

View file

@ -130,6 +130,9 @@ df2 = split_tts(df
all(df2['X'].columns.isin(['gene_name'])) # should be False all(df2['X'].columns.isin(['gene_name'])) # should be False
import pandas as pd
from sklearn.utils import all_estimators from sklearn.utils import all_estimators
all_clfs = all_estimators(type_filter="classifier") all_clfs = all_estimators(type_filter="classifier")
df = pd.DataFrame (all_clfs, columns = ['classifier_name', 'classifier_fn'])
df.to_csv("Model_names_ALL.csv")