HoriaPiv commited on
Commit
7223d1a
·
verified ·
1 Parent(s): cbea050
Files changed (3) hide show
  1. model.joblib +3 -0
  2. predict.py +9 -0
  3. train.py +15 -0
model.joblib ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ddea809ff33bc337eb5c7a4976204c44367545ab280cda551bf33c5b3b18ff7b
3
+ size 863
predict.py ADDED
@@ -0,0 +1,9 @@
 
 
 
 
 
 
 
 
 
 
1
+ import joblib
2
+ import sys
3
+
4
+ model = joblib.load("model.joblib")
5
+
6
+ value = int(sys.argv[1])
7
+ prediction = model.predict([[value]])[0]
8
+
9
+ print("odd" if prediction == 1 else "even")
train.py ADDED
@@ -0,0 +1,15 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import joblib
2
+ from sklearn.linear_model import LogisticRegression
3
+ import numpy as np
4
+
5
+ # Training data (toy example)
6
+ X = np.array([[0], [1], [2], [3], [4], [5]])
7
+ y = np.array([0, 1, 0, 1, 0, 1]) # 0 = even, 1 = odd
8
+
9
+ model = LogisticRegression()
10
+ model.fit(X, y)
11
+
12
+ # Save model
13
+ joblib.dump(model, "model.joblib")
14
+
15
+ print("Model trained and saved as model.joblib")