diff --git a/JenkinsfileDL b/JenkinsfileDL index e261957..f8e83d6 100644 --- a/JenkinsfileDL +++ b/JenkinsfileDL @@ -11,7 +11,7 @@ pipeline { stages { stage('Preparation') { steps { - sh 'pip install pandas tensorflow scikit-learn imbalanced-learn sacred' + sh 'pip install pandas tensorflow scikit-learn imbalanced-learn sacred pymongo' } } stage('Pobierz dane') { diff --git a/JenkinsfileEvaluate b/JenkinsfileEvaluate index e3b43d0..16a5816 100644 --- a/JenkinsfileEvaluate +++ b/JenkinsfileEvaluate @@ -23,7 +23,7 @@ pipeline { } stage('Przygotowanie') { steps { - sh 'pip install pandas tensorflow scikit-learn matplotlib sacred' + sh 'pip install pandas tensorflow scikit-learn matplotlib sacred pymongo' } } stage('Ewaluuj model') { diff --git a/train.py b/train.py index ab1268f..b67d1ea 100644 --- a/train.py +++ b/train.py @@ -4,6 +4,7 @@ from sacred.observers import MongoObserver, FileStorageObserver ex = Experiment('s487187-training') ex.observers.append(MongoObserver(url='mongodb://admin:IUM_2021@172.17.0.1:27017', db_name='sacred')) ex.observers.append(FileStorageObserver('results')) +ex.use_git = False @ex.config def my_config():