From 7ea299d27128d6fedbb49a3f23269027c4ec245c Mon Sep 17 00:00:00 2001 From: Kornelia Girejko Date: Mon, 9 May 2022 11:13:01 +0200 Subject: [PATCH] Zaktualizuj 'train-sacred.py' --- train-sacred.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/train-sacred.py b/train-sacred.py index 4cdac92..1b5d261 100644 --- a/train-sacred.py +++ b/train-sacred.py @@ -16,6 +16,8 @@ from sacred.observers import MongoObserver from sklearn.metrics import accuracy_score from sacred.observers.file_storage import file_storage_option from sacred.observers.mongo import mongo_db_option +from sklearn.metrics import mean_squared_error +from sklearn.metrics import mean_absolute_error