tanu/ML_AI_training
1
0
Fork 0
Code Issues Pull requests Releases Wiki Activity

replaced pipe_abc with pipe_rc related to the same

Browse source
This commit is contained in:
Tanushree Tunstall 2022-05-24 11:57:54 +01:00
parent a4124986c0
commit cdddbde7ec
2 changed files with 2 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

3
classification_params_FS.py
View file

@ -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'))