Skip to content

Commit 1a492b6

Browse files
kaushikmitrBenjaminBraunDev
authored andcommitted
add xgboost regressor and update tpot sampling logic
1 parent 9bbcb9f commit 1a492b6

File tree

12 files changed

+3070
-609
lines changed

12 files changed

+3070
-609
lines changed

config/manifests/inferencepool-resources.yaml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -11,7 +11,7 @@ metadata:
1111
name: latency-predictor-config
1212
namespace: default
1313
data:
14-
LATENCY_RETRAINING_INTERVAL_SEC: "10"
14+
LATENCY_RETRAINING_INTERVAL_SEC: "5"
1515
LATENCY_MIN_SAMPLES_FOR_RETRAIN: "100"
1616
LATENCY_TTFT_MODEL_PATH: "/models/ttft.joblib"
1717
LATENCY_TPOT_MODEL_PATH: "/models/tpot.joblib"

latencypredictor/manifests/latencypredictor_manifest.yaml

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -14,6 +14,8 @@ data:
1414
LATENCY_TPOT_MODEL_PATH: "/models/tpot.joblib"
1515
LATENCY_TTFT_SCALER_PATH: "/models/ttft_scaler.joblib"
1616
LATENCY_TPOT_SCALER_PATH: "/models/tpot_scaler.joblib"
17+
LATENCY_MODEL_TYPE: "xgboost" # or "xgboost"
18+
1719

1820
---
1921
# --- 2. Deployment ---

latencypredictor/requirements.txt

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6,4 +6,5 @@ pandas
66
joblib
77
river
88
pydantic
9-
requests
9+
requests
10+
xgboost

latencypredictor/server.py

Lines changed: 398 additions & 160 deletions
Large diffs are not rendered by default.

0 commit comments

Comments
 (0)