Enric Perpinyà Pitarch
commited on
Commit
•
509a137
1
Parent(s):
a715e4b
new run
Browse files
app.py
CHANGED
@@ -19,29 +19,17 @@ print("Model Loaded...")
|
|
19 |
def wine(type,
|
20 |
fixed_acidity,
|
21 |
volatile_acidity,
|
22 |
-
citric_acid,
|
23 |
-
residual_sugar,
|
24 |
-
chlorides,
|
25 |
-
free_sulfur_dioxide,
|
26 |
-
total_sulfur_dioxide,
|
27 |
-
density,
|
28 |
-
ph,
|
29 |
sulphates,
|
30 |
-
alcohol
|
|
|
31 |
print("Lets taste wine?")
|
32 |
|
33 |
-
df = pd.DataFrame([[type, fixed_acidity, volatile_acidity,
|
34 |
-
citric_acid, residual_sugar,
|
35 |
-
chlorides, free_sulfur_dioxide,
|
36 |
-
total_sulfur_dioxide,
|
37 |
-
density, ph, sulphates, alcohol]],
|
38 |
columns = ['type', 'fixed_acidity', 'volatile_acidity',
|
39 |
-
'
|
40 |
-
'free_sulfur_dioxide', 'total_sulfur_dioxide', 'density',
|
41 |
-
'ph', 'sulphates', 'alcohol'])
|
42 |
|
43 |
print("Predicting...")
|
44 |
-
|
45 |
res = model.predict(df)
|
46 |
print(res)
|
47 |
|
|
|
19 |
def wine(type,
|
20 |
fixed_acidity,
|
21 |
volatile_acidity,
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
22 |
sulphates,
|
23 |
+
alcohol,
|
24 |
+
density):
|
25 |
print("Lets taste wine?")
|
26 |
|
27 |
+
df = pd.DataFrame([[type, fixed_acidity, volatile_acidity, sulphates, alcohol, density]],
|
|
|
|
|
|
|
|
|
28 |
columns = ['type', 'fixed_acidity', 'volatile_acidity',
|
29 |
+
'sulphates', 'alcohol', 'density'])
|
|
|
|
|
30 |
|
31 |
print("Predicting...")
|
32 |
+
print(df.head())
|
33 |
res = model.predict(df)
|
34 |
print(res)
|
35 |
|