Feat: add modules

#54
boostcampaitech7 · Dec 29, 2024 · 4400719 · 4400719
1 parent 652aa27
commit 4400719
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/src/train/EASE_train.py b/src/train/EASE_train.py
@@ -1,7 +1,7 @@
 import numpy as np
 from src.models import EASE
 import pandas as pd
-import mean_squared_error
+from sklearn.metrics import mean_squared_error
 class EASETrainer:
     def __init__(self, args,data):
         """

diff --git a/src/train/SLIM_train.py b/src/train/SLIM_train.py
@@ -3,6 +3,7 @@
 import torch
 import numpy as np
 import pandas as pd
+from sklearn.metrics import mean_squared_error
 
 class SLIMTrainer:
     def __init__(self, args,data):