.rst .pdf

repository open issue suggest edit

callbacks

import tensorwaves.optimizer.callbacks

Collection of loggers that can be inserted into an optimizer as callback.

class CSVSummary(filename: str, function_call_step_size: int = 1, iteration_step_size: Optional[int] = None)

Bases: tensorwaves.optimizer.callbacks.Callback, tensorwaves.optimizer.callbacks.Loadable

static load_latest_parameters(filename: str) → dict

on_function_call_end(function_call: int, logs: Optional[Dict[str, Any]] = None) → None

on_iteration_end(iteration: int, logs: Optional[Dict[str, Any]] = None) → None

on_optimize_end(logs: Optional[Dict[str, Any]] = None) → None

on_optimize_start(logs: Optional[Dict[str, Any]] = None) → None

class Callback

Bases: abc.ABC

Interface for callbacks such as CSVSummary.

See also

Custom callbacks

abstract on_function_call_end(function_call: int, logs: Optional[Dict[str, Any]] = None) → None

abstract on_iteration_end(iteration: int, logs: Optional[Dict[str, Any]] = None) → None

abstract on_optimize_end(logs: Optional[Dict[str, Any]] = None) → None

abstract on_optimize_start(logs: Optional[Dict[str, Any]] = None) → None

class CallbackList(callbacks: Iterable[tensorwaves.optimizer.callbacks.Callback])

Bases: tensorwaves.optimizer.callbacks.Callback

Class for combining Callback s.

Combine different Callback classes in to a chain as follows:

>>> from tensorwaves.optimizer.callbacks import (
...     CallbackList, TFSummary, YAMLSummary
... )
>>> from tensorwaves.optimizer.minuit import Minuit2
>>> optimizer = Minuit2(
...     callback=CallbackList([TFSummary(), YAMLSummary("result.yml")])
... )

on_function_call_end(function_call: int, logs: Optional[Dict[str, Any]] = None) → None

on_iteration_end(iteration: int, logs: Optional[Dict[str, Any]] = None) → None

on_optimize_end(logs: Optional[Dict[str, Any]] = None) → None

on_optimize_start(logs: Optional[Dict[str, Any]] = None) → None

class Loadable

Bases: abc.ABC

abstract static load_latest_parameters(filename: str) → dict

class TFSummary(logdir: str = 'logs', step_size: int = 10, subdir: Optional[str] = None)

Bases: tensorwaves.optimizer.callbacks.Callback

on_function_call_end(function_call: int, logs: Optional[Dict[str, Any]] = None) → None

on_iteration_end(iteration: int, logs: Optional[Dict[str, Any]] = None) → None

on_optimize_end(logs: Optional[Dict[str, Any]] = None) → None

on_optimize_start(logs: Optional[Dict[str, Any]] = None) → None

class YAMLSummary(filename: str, step_size: int = 10)

Bases: tensorwaves.optimizer.callbacks.Callback, tensorwaves.optimizer.callbacks.Loadable

static load_latest_parameters(filename: str) → dict

on_function_call_end(function_call: int, logs: Optional[Dict[str, Any]] = None) → None

on_iteration_end(iteration: int, logs: Optional[Dict[str, Any]] = None) → None

on_optimize_end(logs: Optional[Dict[str, Any]] = None) → None

on_optimize_start(logs: Optional[Dict[str, Any]] = None) → None

optimizer minuit