commandline

Documentation

Overview

Package commandline contains convenience UI training tools for the command line.

Index

• Constants
• func AttachProgressBar(loop *train.Loop)
• func CreateContextSettingsFlag(ctx *context.Context, name string) *string
• func ParseContextSettings(ctx *context.Context, settings string) error
• 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 CreateContextSettingsFlag

func CreateContextSettingsFlag(ctx *context.Context, name string) *string

CreateContextSettingsFlag create a string flag with the given name (if empty it will be named "set") and with a description of the current defined parameters in the context `ctx`.

The flag should be created before the call to `flags.Parse()`.

Example:

func main() {
	ctx := createDefaultContext()
	settings := CreateContextSettingsFlag(ctx, "")
	flags.Parse()
	err := ParseContextSettings(ctx, *settings)
	if err != nil {...}
	...
}

func ParseContextSettings

func ParseContextSettings(ctx *context.Context, settings string) error

ParseContextSettings from for example a flag definition.

The settings are a list separated by ";": e.g.: "param1=value1;param2=value2;...".

All the parameters "param1", "param2", etc. must be already set with default values in the context `ctx`. The default values are also used to set the type to which the string values will be parsed to.

It updates `ctx` parameters accordingly, and returns an error in case a parameter is unknown or the parsing failed.

Note, one can also provide a scope for the parameters: "layer_1/l2_regularization=0.1" will work, as long as a default "l2_regularization" is defined in `ctx`.

See example in CreateContextSettingsFlag, which will create a flag for the settings.

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.