dschandra commited on
Commit
2ff670a
·
verified ·
1 Parent(s): 41660fe

Delete lapse_model.pkl

Browse files
Files changed (1) hide show
  1. lapse_model.pkl +0 -25
lapse_model.pkl DELETED
@@ -1,25 +0,0 @@
1
- import pandas as pd
2
- import xgboost as xgb
3
- import pickle
4
-
5
- # Example dummy dataset
6
- data = {
7
- "policy_term": [10, 15, 20],
8
- "policy_age": [2, 5, 7],
9
- "payment_mode_encoded": [0, 1, 2],
10
- "communication_score": [0.3, 0.7, 0.5],
11
- "lapse_label": [1, 0, 1]
12
- }
13
- df = pd.DataFrame(data)
14
-
15
- # Split features and label
16
- X = df[["policy_term", "policy_age", "payment_mode_encoded", "communication_score"]]
17
- y = df["lapse_label"]
18
-
19
- # Train model
20
- model = xgb.XGBClassifier()
21
- model.fit(X, y)
22
-
23
- # Save the model
24
- with open("lapse_model.pkl", "wb") as f:
25
- pickle.dump(model, f)