Icon

BBC Documents multiclass classification with BERT extension

BBC Documents classification with BERT extension
This workflow trains a multiclass classifier using the Redfield BERT Nodes extension based on Google's BERT model. It then assigns text exerpts with one ofthe five categories: business, entertainment, politics, sport, tech. The BBC data set can be found on Kaggle: https://www.kaggle.com/shivamkushwaha/bbc-full-text-document-classificationFor more information see the workflow metadata. Find it here: View -> Description Pre-processing: - lowercase the input text- split data to training, validation and test samples Data upload: zip archivefrom https://www.kaggle.com/shivamkushwaha/bbc-full-text-document-classification BERT model upload Training the classifier Prediction of categories based either on newly trainedmodel or on the saved and uploaded model Performance analysis andcomparison upload BERTmodellower case text80 training20 test80 training20 validationWithout fine tuningWithout fine tuningupload thesaved modelwithoutfine tuningWith fine tuningWith fine tuningsave the modelwithoutfine tuningno fine tuningsave the modelwithfine tuningupload thesaved modelwithfine tuningno fine tuning BERT Model Selector String Manipulation Partitioning Partitioning BERT ClassificationLearner BERT Predictor Model Reader BERT Predictor BERT ClassificationLearner Model Writer Timer Info Upload the data set Model Writer Model Reader Timer Info Evaluation This workflow trains a multiclass classifier using the Redfield BERT Nodes extension based on Google's BERT model. It then assigns text exerpts with one ofthe five categories: business, entertainment, politics, sport, tech. The BBC data set can be found on Kaggle: https://www.kaggle.com/shivamkushwaha/bbc-full-text-document-classificationFor more information see the workflow metadata. Find it here: View -> Description Pre-processing: - lowercase the input text- split data to training, validation and test samples Data upload: zip archivefrom https://www.kaggle.com/shivamkushwaha/bbc-full-text-document-classification BERT model upload Training the classifier Prediction of categories based either on newly trainedmodel or on the saved and uploaded model Performance analysis andcomparison upload BERTmodellower case text80 training20 test80 training20 validationWithout fine tuningWithout fine tuningupload thesaved modelwithoutfine tuningWith fine tuningWith fine tuningsave the modelwithoutfine tuningno fine tuningsave the modelwithfine tuningupload thesaved modelwithfine tuningno fine tuningBERT Model Selector String Manipulation Partitioning Partitioning BERT ClassificationLearner BERT Predictor Model Reader BERT Predictor BERT ClassificationLearner Model Writer Timer Info Upload the data set Model Writer Model Reader Timer Info Evaluation

Nodes

Extensions

Links

Download

To use this workflow in KNIME, download it from the below URL and open it in KNIME:

Download Workflow
Created by: Lada.Rudnitckaia
Created at: 2020-11-19
On NodePit since: 2021-08-31
Last update: 2024-04-25
Created with KNIME version: v4.3.0 nightly
Tags: #BERT#deeplearning#textmining#classification

Deploy, schedule, execute, and monitor your KNIME workflows locally, in the cloud or on-premises – with our brand new NodePit Runner.

Try NodePit Runner!