Hub
Pricing About
NodeNode / Learner

BERT Classification Learner

Community NodesRedfield BERT Nodes
Drag & drop
Like

Trains a text classification model on top of the provided BERT model. The model is extended with 3 layers:

  • GlobalAveragePooling1D layer
  • Dropout layer
  • Dense layer

Besides the typical classification task where every row is assigned a single class it is also possible to train a model for multi-label classification where a row can be assigned multiple labels.

If a validation table is provided, then the model performance is evaluated on that data after every epoch.

Node details

Input ports
  1. Type: se.redfield.bert.nodes.port.BertModelPortObject
    BERT Model
    BERT Model
  2. Type: Table
    Data Table
    Data Table
  3. Type: Table
    Validation Table
    Optional Validation Table
Output ports
  1. Type: se.redfield.bert.nodes.port.BertClassifierPortObject
    BERT Classifier
    BERT Classifier model
  2. Type: Table
    Training statistics
    Statistics of the training process

Extension

The BERT Classification Learner node is part of this extension:

  1. Go to item

Related workflows & nodes

  1. Go to item
  2. Go to item
  3. Go to item

KNIME
Open for Innovation

KNIME AG
Talacker 50
8001 Zurich, Switzerland
  • Software
  • Getting started
  • Documentation
  • Courses + Certification
  • Solutions
  • KNIME Hub
  • KNIME Forum
  • Blog
  • Events
  • Partner
  • Developers
  • KNIME Home
  • Careers
  • Contact us
Download KNIME Analytics Platform Read more about KNIME Business Hub
© 2025 KNIME AG. All rights reserved.
  • Trademarks
  • Imprint
  • Privacy
  • Terms & Conditions
  • Data Processing Agreement
  • Credits