dutch-youth-comment-classifier / src /start_up /start_up_bad_words_rule.py
nosdigitalmedia's picture
Attempt to set up application
4e0f321
raw
history blame
271 Bytes
import pandas as pd
from src.rule_based_system.BadWordRule import BadWordRule
def create_bad_word_rule(path, is_strict):
df = pd.read_csv(path, header=None)
df.columns = ['word']
bad_words = df['word'].values
return BadWordRule(bad_words, is_strict)