diff --git a/sacred_model.py b/sacred_model.py index 6b4942a..c03d6b4 100644 --- a/sacred_model.py +++ b/sacred_model.py @@ -1,13 +1,13 @@ from sacred import Experiment -from sacred.observers import FileStorageObserver +from sacred.observers import FileStorageObserver, MongoObserver import pandas as pd import numpy as np from tensorflow.keras import Sequential from tensorflow.keras.layers import Dense from sklearn.preprocessing import MinMaxScaler -ex = Experiment("car_price_prediction") - +ex = Experiment("464962") +ex.observers.append(MongoObserver(url='mongodb://admin:IUM_2021@172.17.0.1:27017', db_name='sacred')) ex.observers.append(FileStorageObserver('my_runs'))