Mirrors/the-algorithm
2
0
Fork 0
mirror of https://github.com/twitter/the-algorithm.git synced 2024-06-14 15:18:55 +02:00
Code Issues Packages Projects Releases Wiki Activity

NSFW filter adjustment

Browse Source 
adjusted the nsfw filter prediction to be better
This commit is contained in:
crckrbrrs 2023-05-04 19:55:03 +03:00 committed by GitHub
parent 90d7ea370e
commit 8c9f977247
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
trust_and_safety_models/nsfw/nsfw_text.py
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))