Update nsfw_text.py

Added eggplant emoji counterpart for equality.
This commit is contained in:
Anthony Nguyen 2023-03-31 23:08:25 -04:00 committed by GitHub
parent ec83d01dca
commit 9ed9b3ad5c
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -134,7 +134,7 @@ history = model.fit(
steps_per_epoch=len(ds_train)
)
model.predict(["xxx 🍑"])
model.predict(["xxx 🍑 🍆"])
preds = X_val.processed_text.apply(apply_model)
print(classification_report(y_val, preds >= 0.90, digits=4))