Commit
·
bfd5973
1
Parent(s):
206327a
Update main.py
Browse files
main.py
CHANGED
@@ -38,10 +38,10 @@ async def predict(input: InputFeatures):
|
|
38 |
'Blood_Work_Result_4', 'patients_age']]
|
39 |
|
40 |
#creating pipelines
|
41 |
-
num_pipeline= Pipeline([('imputer', SimpleImputer()),('scaler', StandardScaler())])
|
42 |
|
43 |
|
44 |
-
full_pipeline=ColumnTransformer([('num_pipe',num_pipeline,num_attr)])
|
45 |
|
46 |
XGB = Pipeline([
|
47 |
("col_trans", full_pipeline), # You need to define full_pipeline
|
@@ -55,8 +55,8 @@ full_pipeline=ColumnTransformer([('num_pipe',num_pipeline,num_attr)])
|
|
55 |
|
56 |
return prediction
|
57 |
|
58 |
-
|
59 |
-
|
60 |
|
61 |
|
62 |
|
|
|
38 |
'Blood_Work_Result_4', 'patients_age']]
|
39 |
|
40 |
#creating pipelines
|
41 |
+
num_pipeline= Pipeline([('imputer', SimpleImputer()),('scaler', StandardScaler())])
|
42 |
|
43 |
|
44 |
+
full_pipeline=ColumnTransformer([('num_pipe',num_pipeline,num_attr)])
|
45 |
|
46 |
XGB = Pipeline([
|
47 |
("col_trans", full_pipeline), # You need to define full_pipeline
|
|
|
55 |
|
56 |
return prediction
|
57 |
|
58 |
+
if __name__ == '__main__':
|
59 |
+
uvicorn.run("Main:app", reload=True)
|
60 |
|
61 |
|
62 |
|