fix import

snuhcs-course · Nov 2, 2023 · 91f215a · 91f215a
1 parent f4d6a96
commit 91f215a
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 4 deletions.
diff --git a/ml/review/model.py b/ml/review/model.py
@@ -3,7 +3,7 @@
 from torch import nn
 from torch.utils.data import Dataset
 
-from review.tokenization import BERTSentenceTransform
+from tokenization import BERTSentenceTransform
 
 
 class BERTClassifier(nn.Module):

diff --git a/ml/review/predict.py b/ml/review/predict.py
@@ -2,10 +2,10 @@
 import numpy as np
 import torch
 from kobert_tokenizer import KoBERTTokenizer
-from review.model import BERTDataset
+from model import BERTDataset
 
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
-PATH = ""
+PATH = "../model"
 
 
 def predict(predict_sentence):

diff --git a/ml/review/train.py b/ml/review/train.py
@@ -9,7 +9,7 @@
 from transformers import AdamW, BertModel
 from transformers.optimization import get_cosine_schedule_with_warmup
 
-from review.model import BERTClassifier, BERTDataset
+from model import BERTClassifier, BERTDataset
 
 data = pd.read_csv("review_data.csv")