Image Classifier (Experimental)

Skorch Binary Image Classifier https://skorch.readthedocs.io/en/stable/classifier.html

Documentation

Attributes

Model training history

Model training history

Definition

Output ports

model model

Model

Configuration

Batch size (batch_size)

Mini-batch size. If batch size is -1, a single batch with all the data will be used during training and validation.

Cross validation (cross_validation)

Cross validation

Early stopping (early_stopping)

Early stopping

Learning rate (lr)

Learning rate

Maximum number of epochs (max_epochs)

The number of epochs to train for each fit.

Optimizer (optimizer)

The optimizer (update rule) used to optimize the module

Implementation

class node_neuralnetwork.BinaryImageClassifier