This commit is contained in:
eugene 2023-06-07 00:46:30 +02:00
parent 4ba5294f0b
commit 31a651859b

View File

@ -5,6 +5,7 @@ import torch.optim as optim
from torch.utils.data import DataLoader, Dataset from torch.utils.data import DataLoader, Dataset
from sklearn.model_selection import train_test_split from sklearn.model_selection import train_test_split
from sklearn.preprocessing import LabelEncoder from sklearn.preprocessing import LabelEncoder
from sklearn.metrics import mean_squared_error
import pickle import pickle
from sacred import Experiment from sacred import Experiment
from sacred.observers import FileStorageObserver, MongoObserver from sacred.observers import FileStorageObserver, MongoObserver