Spaces:
Runtime error
Runtime error
Commit ·
0184946
1
Parent(s): b2be161
updated
Browse files- model.pkl +3 -0
- model.py +39 -0
- requirements.txt +6 -6
model.pkl
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:b9123d8808c50fe91e2aeb83907ef24a5a0bf70f6b163591cd79991c35187605
|
| 3 |
+
size 419
|
model.py
ADDED
|
@@ -0,0 +1,39 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
import numpy as np
|
| 2 |
+
import pandas as pd
|
| 3 |
+
from sklearn.model_selection import train_test_split
|
| 4 |
+
from sklearn.linear_model import LinearRegression
|
| 5 |
+
import pickle
|
| 6 |
+
|
| 7 |
+
|
| 8 |
+
# Importing the dataset
|
| 9 |
+
# dataset = pd.read_csv('dataset/Sales_Salary_Data.csv')
|
| 10 |
+
dataset = pd.read_csv('dataset/Sales_Salary_Data_IDR.csv')
|
| 11 |
+
|
| 12 |
+
# seprate feature & target
|
| 13 |
+
X = dataset.iloc[:, :-1].values
|
| 14 |
+
y = dataset.iloc[:, 1].values
|
| 15 |
+
|
| 16 |
+
# Splitting the dataset into the Training set and Test set
|
| 17 |
+
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size = 0.25, random_state = 0)
|
| 18 |
+
|
| 19 |
+
|
| 20 |
+
# Fitting Simple Linear Regression to the Training set
|
| 21 |
+
regressor = LinearRegression()
|
| 22 |
+
regressor.fit(X_train, y_train)
|
| 23 |
+
|
| 24 |
+
|
| 25 |
+
# Predicting the Test set results
|
| 26 |
+
y_pred = regressor.predict(X_test)
|
| 27 |
+
|
| 28 |
+
# Saving serialized model to disk
|
| 29 |
+
pickle.dump(regressor, open('model.pkl','wb'))
|
| 30 |
+
#joblib.dump(regressor, 'model.pkl')
|
| 31 |
+
|
| 32 |
+
|
| 33 |
+
# Loading model to compare the results
|
| 34 |
+
model = pickle.load(open('model.pkl','rb'))
|
| 35 |
+
#model = joblib.load('model.pkl')
|
| 36 |
+
|
| 37 |
+
print("Regressor model output", regressor.predict([[1.8]]))
|
| 38 |
+
print("Saved model output", model.predict([[1.8]]))
|
| 39 |
+
|
requirements.txt
CHANGED
|
@@ -1,9 +1,9 @@
|
|
| 1 |
-
joblib==
|
| 2 |
-
Flask==2.
|
| 3 |
-
numpy==1.
|
| 4 |
-
pandas==
|
| 5 |
-
scikit_learn==
|
| 6 |
gunicorn==19.9.0
|
| 7 |
|
| 8 |
uvicorn[standard]==0.17.*
|
| 9 |
-
requests==2.
|
|
|
|
| 1 |
+
joblib==1.1.
|
| 2 |
+
Flask==2.2.2
|
| 3 |
+
numpy==1.23.5
|
| 4 |
+
pandas==1.5.3
|
| 5 |
+
scikit_learn==1.2.1
|
| 6 |
gunicorn==19.9.0
|
| 7 |
|
| 8 |
uvicorn[standard]==0.17.*
|
| 9 |
+
requests==2.28.1
|