dreem

dreem

Top-level package for dreem.

Modules:

Name	Description
cli	This module contains the command line interfaces for the dreem package.
datasets	Data loading and preprocessing.
inference	Tracking Inference using GTR Model.
io	Module containing input/output data structures for easy storage and manipulation.
models	Model architectures and layers.
training	Initialize training module.
version	Central location for version information.

Functions:

Name	Description
setup_logging	Setup logging based on logging.yaml.

setup_logging()

Setup logging based on logging.yaml.

Source code in dreem/__init__.py

def setup_logging():
    """Setup logging based on `logging.yaml`."""
    import logging
    import yaml
    import os

    package_directory = os.path.dirname(os.path.abspath(__file__))

    with open(os.path.join(package_directory, "..", "logging.yaml"), "r") as stream:
        logging_cfg = yaml.load(stream, Loader=yaml.FullLoader)

    logging.config.dictConfig(logging_cfg)
    logger = logging.getLogger("dreem")