zakariae ayougil commited on
Commit ·
a05a6ba
1
Parent(s): e0f788b
Initial commit: upload traffic volume model and sample inference script
Browse files- inference.py +8 -0
- 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
|