0 ×

BERT Classification Learner

BERT extension for KNIME Workbench version 0.0.1.v202012081121 by Redfield AB

The node uses BERT model and adds a predefined neural network on top. There are 3 layers added:

  • GlobalAveragePooling1D layer
  • Dropout layer
  • Dense layer
The trained model can be applied for multi-class classification.

Options

Settings

Sentence column
A column with a plain text (String), that contains text to be classified. No special pre-processing is needed.
Class column
A column that contains class labels.
Max sequence length
The maximum length of a sequence after tokenization, limit is 512.

Advanced

Number of epochs
The number of epochs used for training the classifier.
Batch size
The size of a chunk of the input data used for model update.
Fine tune BERT
If checked than BERT model will be trained along with the additional classifier. It takes longer time to fine tune BERT, but the results are usually better.
Optimizer
Available optimizers and their configuration.

Input Ports

Icon
BERT Model
Icon
Data Table
Icon
Validation Table

Output Ports

Icon
BERT Classifier model
Icon
Stats

Best Friends (Incoming)

Best Friends (Outgoing)

Workflows

Installation

To use this node in KNIME, install Redfield BERT Nodes from the following update site:

KNIME 4.3

A zipped version of the software site can be downloaded here.

You don't know what to do with this link? Read our NodePit Product and Node Installation Guide that explains you in detail how to install nodes to your KNIME Analytics Platform.

Wait a sec! You want to explore and install nodes even faster? We highly recommend our NodePit for KNIME extension for your KNIME Analytics Platform. Browse NodePit from within KNIME, install nodes with just one click and share your workflows with NodePit Space.

Developers

You want to see the source code for this node? Click the following button and we’ll use our super-powers to find it for you.