Mustafa Al Hamad commited on
Commit
10a3892
·
1 Parent(s): 9fb7507

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -6,7 +6,7 @@ from docx import Document
6
  from docx.shared import Inches
7
  from docx.dml.color import ColorFormat
8
  import sklearn
9
-
10
  import numpy as np
11
  import pandas as pd
12
  from sklearn.linear_model import LogisticRegression
@@ -140,10 +140,11 @@ def greet(operation,filer):
140
  kf = StratifiedKFold(n_splits=10)
141
  models1 = [KNeighborsClassifier(leaf_size=10,metric='manhattan'),
142
  RandomForestClassifier(max_depth=100),
 
143
  VotingClassifier(estimators=[('knn',
144
  KNeighborsClassifier(leaf_size=10,
145
  metric='manhattan')),
146
- ('rf', RandomForestClassifier(max_depth=100))])]
147
  metrics = dict()
148
  for model in models1:
149
  model.fit(X_smote,y_smote)
 
6
  from docx.shared import Inches
7
  from docx.dml.color import ColorFormat
8
  import sklearn
9
+ from lightgbm import LGBMClassifier
10
  import numpy as np
11
  import pandas as pd
12
  from sklearn.linear_model import LogisticRegression
 
140
  kf = StratifiedKFold(n_splits=10)
141
  models1 = [KNeighborsClassifier(leaf_size=10,metric='manhattan'),
142
  RandomForestClassifier(max_depth=100),
143
+ LGBMClassifier(n_estimators=200, num_leaves=60),
144
  VotingClassifier(estimators=[('knn',
145
  KNeighborsClassifier(leaf_size=10,
146
  metric='manhattan')),
147
+ ('rf', RandomForestClassifier(max_depth=100)),('gm',LGBMClassifier(n_estimators=200, num_leaves=60))])]
148
  metrics = dict()
149
  for model in models1:
150
  model.fit(X_smote,y_smote)