toasty-dust-16
Tobiasuruali's workspace
1 | - |
| |
1 | + |
| |
2 | - |
| |
2 | + |
| |
3 | - |
| |
3 | + |
| |
4 | - |
| |
4 | + |
| |
5 | - |
| |
5 | + |
| |
6 | 6 | ||
7 | 7 | if __name__ == "__main__": | |
8 | 8 | data_preparation.initialize_wandb() | |
9 | 9 | num_classes, input_shape, x_train, y_train, x_test, y_test = data_preparation.prepare_data() | |
10 | 10 |
| |
11 | - |
| |
11 | + | model = build.build_model(num_classes, input_shape) | |
12 | - |
| |
12 | + | build.train_model(x_train, y_train, x_test, y_test, model) | |
13 | - |
| |
13 | + | inspection.save_model(model) | |
14 | 14 |
| |
15 | 15 | loaded_model = inspection.load_model() | |
16 | 16 | inspection.evaluate_loaded_model(x_test, y_test, loaded_model) | |
17 | 17 |
| |
18 | 18 | database_pg.create_milestone3_db() | |
19 | 19 | database_pg.create_input_pred_db() | |
20 | 20 | random_img_x, squeezed_random_img_x, img_from_db = database_pg.insert_load_random_image() | |
21 | 21 | database_pg.predict_and_persist(img_from_db, loaded_model) |