diff --git a/.dvc/.gitignore b/.dvc/.gitignore new file mode 100644 index 0000000..528f30c --- /dev/null +++ b/.dvc/.gitignore @@ -0,0 +1,3 @@ +/config.local +/tmp +/cache diff --git a/.dvc/config b/.dvc/config new file mode 100644 index 0000000..e69de29 diff --git a/.dvcignore b/.dvcignore new file mode 100644 index 0000000..5197305 --- /dev/null +++ b/.dvcignore @@ -0,0 +1,3 @@ +# Add patterns of files dvc should ignore, which could improve +# the performance. Learn more at +# https://dvc.org/doc/user-guide/dvcignore diff --git a/ml_training.py b/ml_training.py index 2e8b4cb..475bd7b 100644 --- a/ml_training.py +++ b/ml_training.py @@ -5,10 +5,12 @@ from sklearn.model_selection import train_test_split from sklearn.preprocessing import StandardScaler import sys import sacred -from sacred.observers import FileStorageObserver +from sacred.observers import FileStorageObserver, MongoObserver ex = sacred.Experiment("Training model") ex.observers.append(FileStorageObserver('training_experiment')) +# ex.observers.append(MongoObserver(url='mongodb://admin:IUM_2021@172.17.0.1:27017', +# db_name='sacred')) @ex.config