Text Classification Using BERT & Tensorflow

In this article, we are going to implement an email class classification whether it is spam or nonspam using BERT. Columns 2,3,4 contain no important data and can be deleted. Also, we rename column v1 as "category" and v2 as "message". Here we have 4825 ham and 747 spam emails. Creating a new column, if the value is spam the value of this spam column will be 1 and for ham, it will be zero.