diff --git a/UQ_FS_fn.py b/UQ_FS_fn.py index dc09cf3..2fdba38 100644 --- a/UQ_FS_fn.py +++ b/UQ_FS_fn.py @@ -19,7 +19,7 @@ def fsgs(input_df , var_type = ['numerical' , 'categorical' , 'mixed'] - , fs_estimator = [LogisticRegression(**rs)] + #, fs_estimator = [LogisticRegression(**rs)] , fs = RFECV(DecisionTreeClassifier(**rs) , cv = StratifiedKFold(n_splits = 10 , shuffle = True,**rs) diff --git a/classification_params_FS.py b/classification_params_FS.py index 12fcaaf..0991e25 100644 --- a/classification_params_FS.py +++ b/classification_params_FS.py @@ -145,7 +145,6 @@ param_grid_gbc = [ { # 'clf': [GradientBoostingClassifier(**rs)], 'clf__n_estimators' : [10, 100, 200, 500, 1000] - , 'clf__n_estimators' : [10, 100, 1000] , 'clf__learning_rate': [0.001, 0.01, 0.1] , 'clf__subsample' : [0.5, 0.7, 1.0] , 'clf__max_depth' : [3, 7, 9] @@ -354,7 +353,7 @@ param_grid_qda = [ estimator = RidgeClassifier(**rs) # Define pipleline with steps -pipe_abc = Pipeline([ +pipe_rc = Pipeline([ ('pre', MinMaxScaler()) , ('fs', RFECV(DecisionTreeClassifier(**rs), cv = cv, scoring = 'matthews_corrcoef')) # , ('fs', RFECV(estimator, cv = cv, scoring = 'matthews_corrcoef'))