zakariae ayougil commited on
Commit
a05a6ba
·
1 Parent(s): e0f788b

Initial commit: upload traffic volume model and sample inference script

Browse files
Files changed (2) hide show
  1. inference.py +8 -0
  2. traffic_volume_model.joblib +3 -0
inference.py ADDED
@@ -0,0 +1,8 @@
 
 
 
 
 
 
 
 
 
1
+ import pandas as pd
2
+ import joblib
3
+
4
+ def predict_volume(input_data):
5
+ model = joblib.load("traffic_volume_model.joblib")
6
+ max_volume = 7280
7
+ prediction = model.predict(input_data) * max_volume
8
+ return prediction
traffic_volume_model.joblib ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a22dfd3bd2350574656a895828ee3e3ce80def70273d6cfb387b01aecb54d61f
3
+ size 335434337