prateekjoshi565 / Fine-Tuning-BERT

Apache License 2.0
139 stars 119 forks source link

Add AUC score for AUC-ROC curve #8

Open ps3-app opened 3 years ago

ps3-app commented 3 years ago

How do I generate AUC score for AUC-ROC curve I tried the predict_proba but its give me an error :"'BERT_Arch' object has no attribute 'predict_proba'"

furthermore, is it possible to add other neural network layer on top of Bert such as LSTM or BiLSTM.