Our goal in this Demo is to use the pre trained BERT model to complete the task of classifying items on e-commerce websites, and to judge whether the item belongs to the Fashion industry by entering the title of the item.
The model training for an epoch has converged well. The following indicators are Accuracy, AUC, Precision, Recall, F1, etc. If we take Threshold=0.5, then