ml_logger.logger package

Submodules

ml_logger.logger.base module

Abstract logger class.

class ml_logger.logger.base.Logger(config: Dict[str, Any])

Bases: object

Abstract Logger Class.

abstract write(log: Dict[str, Any]) → None

Interface to write the log.

Parameters

log (LogType) – Log to write

ml_logger.logger.filesystem module

Functions to interface with the filesystem.

class ml_logger.logger.filesystem.Logger(config: Dict[str, Any])

Bases: ml_logger.logger.base.Logger

Logger class that writes to the filesystem.

write(log: Dict[str, Any]) → None

Write the log to the filesystem.

Parameters

log (LogType) – Log to write

ml_logger.logger.filesystem.get_logger_file_path(logger_dir: str, filename: Optional[str], filename_prefix: str, filename_suffix: str) → str

Get path to the file (to write logs to).

ml_logger.logger.filesystem.to_json_serializable(val: Any) → Any

Serialize values as json.

ml_logger.logger.mlflow module

Logger class that writes to mlflow.

class ml_logger.logger.mlflow.Logger(config: Dict[str, Any])

Bases: ml_logger.logger.base.Logger

Logger class that writes to mlflow.

write(log: Dict[str, Any]) → None

Write the log to mlflow.

Parameters

log (LogType) – Log to write

write_config(config: Dict[str, Any]) → None

Write the config to mlflow.

Parameters

config (ConfigType) – Config to write

write_metric(metric: Dict[str, Any]) → None

Write metric to mlflow.

Parameters

metric (MetricType) – Metric to write

ml_logger.logger.mongo module

Functions to interface with the mongodb.

class ml_logger.logger.mongo.Logger(config: Dict[str, Any])

Bases: ml_logger.logger.base.Logger

Logger class that writes to the mongodb.

write(log: Dict[str, Any]) → None

Write the log to the filesystem.

Parameters

log (LogType) – Log to write

ml_logger.logger.tensorboard module

Logger class that writes to tensorboard.

class ml_logger.logger.tensorboard.Logger(config: Dict[str, Any])

Bases: ml_logger.logger.base.Logger

Logger class that writes to tensorboardX.

write(log: Dict[str, Any]) → None

Write the log to tensorboard.

Parameters

log (LogType) – Log to write

write_config(config: Dict[str, Any]) → None

Write the config to tensorboard.

Parameters

config (ConfigType) – Config to write

write_metric(metric: Dict[str, Any]) → None

Write metric to tensorboard.

Parameters

metric (MetricType) – Metric to write

ml_logger.logger.wandb module

Logger class that writes to wandb.

class ml_logger.logger.wandb.Logger(config: Dict[str, Any])

Bases: ml_logger.logger.base.Logger

Logger class that writes to wandb.

write(log: Dict[str, Any]) → None

Write log to wandb.

Parameters

log (LogType) – Log to write

write_config(config: Dict[str, Any]) → None

Write config to wandb.

Parameters

config (ConfigType) – Config to write

write_metric(metric: Dict[str, Any]) → None

Write metric to wandb.

Parameters

metric (MetricType) – Metric to write

Module contents