def select_model(self,
selected_model: str = "TextCnn_char",
scrach: bool = True):
'''
[설명]
선택된 모델을 로드
[input]
selected_model(str): 사용할 모델명을 선택
("TextCnn_char", "TextCnn", or "Bert_classification")
scrach: 완전 처음부터 학습 할지 여부를 결정
(True or False)
[예시]
'''
if selected_model == "TextCnn_char":
if scrach:
import Model as wkm
self.model = wkm.TextCnn_char(max_len_voca=self.max_len_voca)
else:
self.model = tf.keras.models.load_model('./path/to/location')
if selected_model == "TextCnn":
if scrach:
import Model as wkm
self.model = wkm.TextCnn(max_len_voca=self.max_len_voca)
else:
self.model = tf.keras.models.load_model('./path/to/location')
if selected_model == "Bert_classification":
import Model as wkm
self.model = wkm.Bert_classification(
model_name='bert-base-multilingual-cased',
dir_path='bert_ckpt',
num_class=2)
