Spaces:
Runtime error
Runtime error
suarkadipa
commited on
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
|