Skip to main content

Tobiasuruali's workspace

1
-
import 
d
ata_preparation 
1
+
import 
co
d
epy.d
ata_preparation 
as data_preparation 
2
-
import 
build_model as build
2
+
import 
codepy.
build_model as build
3
-
import 
model_inspection as inspection
3
+
import 
codepy.
model_inspection as inspection
4
-
import 
p
redictions
4
+
import 
code
p
y.p
redictions
 as predictions
5
-
import 
d
atabase_pg
5
+
import 
co
d
epy.d
atabase_pg
 as database_pg
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
    #Comment form here..
11
-
    
# 
model = build.build_model(num_classes, input_shape)
11
+
    
model = build.build_model(num_classes, input_shape)
12
-
    
# 
build.train_model(x_train, y_train, x_test, y_test, model)
12
+
    
build.train_model(x_train, y_train, x_test, y_test, model)
13
-
    
# 
inspection.save_model(model)
13
+
    
inspection.save_model(model)
14
14
    #..To here for testing purposes 
15
15
    loaded_model = inspection.load_model()
16
16
    inspection.evaluate_loaded_model(x_test, y_test, loaded_model)
17
17
    # predictions.predict_on_data(x_test, y_test, loaded_model)
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)