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