upd

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