모델

https://drive.google.com/drive/folders/1-5fk-xSCmLJE8SWkt7rI_mxhM2AlDQl8?usp=sharing, https://drive.google.com/drive/folders/1-V_yRIoQH9C-2iuA7L7F_KeUd1vQ67EA?usp=sharing

Requirement

pytorch ← conda install pytorch

tensorflow >=2.2

keras

transformers

코드

perfume-reconmendation/bert_classification

함수

# 로드로 한번 로드하여 변수에 등록해놓고 test 함수 이용시 넣어 사용하면 된다.
def load_model(model_path, tokenizer_path):
  return model: BertForSequenceClassification, tokenizer: BertTokenizer

# test 4개의 label로 분류한다.
def test(text, model, tokenizer):
	...
	return int: 0|1|2|3

model과 tokenizer는 src/model src/tokenizer를 이용하면 된다. (폴더형태이다)

perfume-reconmendation/bert_classification

Params