Update README.md
Browse files
README.md
CHANGED
|
@@ -118,7 +118,7 @@ def learning(model, x_train_ten, y_train_ten, x_test_ten, y_test_ten, x_val_ten,
|
|
| 118 |
validation(model, x_val_ten, y_val_ten)
|
| 119 |
|
| 120 |
save_file(model.state_dict(), 'temperature_classifier.safetensors')
|
| 121 |
-
print('
|
| 122 |
|
| 123 |
train_data = {
|
| 124 |
# ===== Class 0: Negative numbers =====
|
|
|
|
| 118 |
validation(model, x_val_ten, y_val_ten)
|
| 119 |
|
| 120 |
save_file(model.state_dict(), 'temperature_classifier.safetensors')
|
| 121 |
+
print('Model temperature_classifier.safetensors saved')
|
| 122 |
|
| 123 |
train_data = {
|
| 124 |
# ===== Class 0: Negative numbers =====
|