BERT Predictor

Applies a BERT Classification model to the input table.

Options

Sentence column: A column with a plain text (String) or Documents, that contains text to be classified. No special pre-processing is needed.
Batch size: The number of rows to feed to the model simultaniously. During prediction it is typically possible to use a larger batch size than during model training.
Change prediction column name: If active a column with provided name will be created in the output table. Otherwise the default name will be used for the column with predictions.
Append individual class probabilities: If active the columns with class probabilities will be created in the output table.
Suffix for probability columns: If not empty the suffix will be added to every column name for class probabilities columns.

Use custom threshold for assigning the classes: If active it is possible to change the probability threshold for assigning the class. By default class is assigned by the biggest probability value.
Probability threshold: The class is assigned if the class probability is equal or higher then the value. Several classes might be assigned.
Fixed number of classes per prediction: If active it is possible to set up the desired number of classes for a prediction.
Number of classes per prediction: The fixed number of classes that will be assigned to the prediction.
Use custom class separator: If active it is possible to enter the separation character for the prediction. Prediction is represented as a string with separated values.
Class separator: A character that separates multiple labels for prediction.

Python

Select one of the Python execution environment options:

use default Python environment for the Redfield BERT Nodes (can be configured on the preference page)
use Conda environment from a Conda flow variable (only selectable if such a flow variable is available)

This node has no views

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.

To use this node in KNIME, install the extension Redfield BERT Nodes from the below update site following our NodePit Product and Node Installation Guide:

v5.2

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

Plugin provider: Redfield AB

Plugin version: 1.0.3.202402210241

On NodePit since: 2023-12-06

Last update: 2024-04-26

KNIME versions: v5.2, v5.1, v4.7, v4.6, v4.5, v4.4, v4.3, v4.2

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