mattia33011 commited on
Commit
f863b58
·
1 Parent(s): b5577ed

fix: refactor local folders

Browse files
Files changed (2) hide show
  1. .gitignore +3 -1
  2. src/main.py +1 -2
.gitignore CHANGED
@@ -1 +1,3 @@
1
- my_models
 
 
 
1
+ my_models
2
+ localscripts
3
+ checkpoint
src/main.py CHANGED
@@ -5,7 +5,7 @@ import numpy as np
5
 
6
  X_test = x_test/255.0
7
 
8
- model = tf.keras.models.load_model('my_models_h5/fashion_model.keras')
9
 
10
  model.evaluate(X_test, y_test)
11
 
@@ -15,4 +15,3 @@ y_pred = np.argmax(y_pred, axis=1)
15
  from sklearn.metrics import classification_report
16
  #report per predire la accuracy
17
  print(classification_report(y_test, y_pred))
18
-
 
5
 
6
  X_test = x_test/255.0
7
 
8
+ model = tf.keras.models.load_model('my_models/fashion_model.keras')
9
 
10
  model.evaluate(X_test, y_test)
11
 
 
15
  from sklearn.metrics import classification_report
16
  #report per predire la accuracy
17
  print(classification_report(y_test, y_pred))