test
Browse files- model.joblib +3 -0
- predict.py +9 -0
- 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")
|