Views
No views yet
mult-label classification을 수행하며, 유해표현이 포함되는지, 유해표현이라면 그 유형을 판단(분류) 합니다.text-classification에 해당합니다.TTA-DQA/hate_sentence입니다."0": insult"1": abuse"2": obscenity"3": TVPC(Threats of violence/promotion of crime)"4": sexuality"5": age"6": race and region"7": disabled"8": religion"9": politics"10": job"11": no_hateTTA-DQA/hate_sentence)5e-60.0150162BertWordPieceTokenizer405MBpytorch ~= 1.8.0transformers ~= 4.0.01from transformers import AutoTokenizer, AutoModelForSequenceClassification, pipeline
2
3model_name = "TTA-DQA/MultiLabel_KrBERT_Medium_Finetuning"
4tokenizer = AutoTokenizer.from_pretrained(model_name)
5model = AutoModelForSequenceClassification.from_pretrained(model_name)
6classifier = pipeline("text-classification", model=model, tokenizer=tokenizer)
7
8sentences = ["오늘 점심 뭐 먹을까?", "이 나쁜 놈아."]
9results = classifier(sentences)'❗ 본 모델의 결과는 절대적인 유해 표현 기준이 아님을 유의해 주세요.