욕남이 (욕설 감지 API) 만들기

프로젝트 주제

Tensorflow keras를 사용하여 욕설 감지 API를 제작해보기로 했다. Django를 사용하여 backend-server를 구축할 예정이고, API 뿐만 아니라 포트폴리오 용으로 API를 체험해 볼 수 있는 frontend 도 간단하게 제작할 예정이다.

AI 학습하기

욕설 데이터셋 : https://github.com/ZIZUN/korean-malicious-comments-dataset/blob/master/Dataset.csv

학습모델

model = keras.Sequential([
    keras.layers.Embedding(15288, 255), 
    keras.layers.GlobalAveragePooling1D(), 
    keras.layers.Dense(32, activation='relu'),
    keras.layers.Dropout(0.1),
    keras.layers.Dense(1, activation='sigmoid')
])


model.compile(optimizer='adam',
              metrics=['accuracy'],
              loss="binary_crossentropy")

학습 결과 (red : training, blue : validation)