config.py

base_model = 'beomi/KcELECTRA-base'

entity_property_pair =['주차#일반', '편의시설#마트', '편의시설#쇼핑', '학군#일반', '학군#학교 접근성', '학군#학원 접근성', 
'학군#양육 환경', '인프라#일반', '인프라#상권', '인프라#교통', '인프라#병원', '인프라#대중교통', '환경#일반', '환경#소음', '편의시설#일반',
'환경#공원', '환경#단지 관리', '환경#뷰', '환경#조경', '환경#관리비', '구조#집 구조', '가격#시세', '전망#일반', '학군#유치원']

entity2str = dict(zip(entity_property_pair, map(lambda x: x.replace("#", ", ").replace("/", ", "), entity_property_pair)))


tf_id_to_name = ['True', 'False']
tf_name_to_id = {tf_id_to_name[i]: i for i in range(len(tf_id_to_name))}

polarity_id_to_name = ['positive', 'negative']
polarity_name_to_id = {polarity_id_to_name[i]: i for i in range(len(polarity_id_to_name))}