commandline

Documentation

Overview

Package commandline contains convenience UI training tools for the command line. In includes a ProgressBar for train.Loop and ReportEval for the registered eval metrics in train.Trainer.

Index

• Constants
• func AttachProgressBar(loop *train.Loop)
• func ReportEval(trainer *train.Trainer, datasets ...train.Dataset) error

Constants

const ProgressBarName = "gomlx.ml.train.commandline.progressBar"

Functions

func AttachProgressBar

func AttachProgressBar(loop *train.Loop)

AttachProgressBar creates a commandline progress bar and attaches it to the Loop, so that everytime Loop is run it will display a progress bar with progression and metrics.

The associated data will be attached to the train.Loop, so nothing is returned.

func ReportEval

func ReportEval(trainer *train.Trainer, datasets ...train.Dataset) error

ReportEval reports on the command line the results of evaluating the datasets using trainer.Eval.