Team Ai
Modelpublic

Johnesss/Toxic-Comment-Classification

sourceHugging Faceupdated 2y agoView on Hugging Face
1likes
Model Card

Model description

This model used for text classification with toxic and non-toxic labels.

Intended uses & limitations

If you want to reuse model, try copy this

from huggingface_hub import from_pretrained_keras
reloaded_model = from_pretrained_keras('Johnesss/Toxic-Comment-Classification')

y_testing=reloaded_model.predict(x_testing,verbose=1,batch_size=32)
test_df['Toxic']=['Not Toxic' if x<0.5 else 'Toxic' for x in y_testing]
test_df[['comment_text','Toxic']].head(20)

Training and evaluation data

Full info in .ipynb file

Training procedure

Training hyperparameters

The following hyperparameters were used during training:

HyperparametersValue
nameAdam
weight_decayNone
clipnormNone
global_clipnormNone
clipvalueNone
use_emaFalse
ema_momentum0.99
emaoverwritefrequencyNone
jit_compileFalse
islegacyoptimizerFalse
learning_rate0.0010000000474974513
beta_10.9
beta_20.999
epsilon1e-07
amsgradFalse
training_precisionfloat32

## Model Plot

<details> <summary>View Model Plot</summary>

[image]

</details>