From 31a651859b64df271cc72ed74a084539613a3eb1 Mon Sep 17 00:00:00 2001 From: eugene Date: Wed, 7 Jun 2023 00:46:30 +0200 Subject: [PATCH] upd --- script7.py | 1 + 1 file changed, 1 insertion(+) diff --git a/script7.py b/script7.py index da31889..0437174 100644 --- a/script7.py +++ b/script7.py @@ -5,6 +5,7 @@ import torch.optim as optim from torch.utils.data import DataLoader, Dataset from sklearn.model_selection import train_test_split from sklearn.preprocessing import LabelEncoder +from sklearn.metrics import mean_squared_error import pickle from sacred import Experiment from sacred.observers import FileStorageObserver, MongoObserver