asrvastava commited on
Commit
cd4f98e
·
1 Parent(s): 34c147f
Files changed (2) hide show
  1. __pycache__/main.cpython-39.pyc +0 -0
  2. main.py +3 -3
__pycache__/main.cpython-39.pyc CHANGED
Binary files a/__pycache__/main.cpython-39.pyc and b/__pycache__/main.cpython-39.pyc differ
 
main.py CHANGED
@@ -36,12 +36,12 @@ def predict(data: PredictRequest):
36
  features = [data.AG_X, data.AG_Y, data.AG_Z, data.Acc_X, data.Acc_Y, data.Acc_Z, data.Gravity_X, data.Gravity_Y,
37
  data.Gravity_Z, data.RR_X, data.RR_Y, data.RR_Z, data.RV_X, data.RV_Y, data.RV_Z, data.cos]
38
 
39
- predictions = loaded_model.predict([features])
40
-
41
 
42
  activities = ["Running", "Sitting", "Standing", "Walking", "DownStairs", "Upstairs"]
43
  activity = activities[predictions]
44
 
45
- return {'activity': predictions}
46
 
47
 
 
36
  features = [data.AG_X, data.AG_Y, data.AG_Z, data.Acc_X, data.Acc_Y, data.Acc_Z, data.Gravity_X, data.Gravity_Y,
37
  data.Gravity_Z, data.RR_X, data.RR_Y, data.RR_Z, data.RV_X, data.RV_Y, data.RV_Z, data.cos]
38
 
39
+ predictions = loaded_model.predict([features])[0]
40
+ print(predictions)
41
 
42
  activities = ["Running", "Sitting", "Standing", "Walking", "DownStairs", "Upstairs"]
43
  activity = activities[predictions]
44
 
45
+ return {'activity': activity}
46
 
47