neural_compressor.ux.components.db_manager

The db_manager package contains all components used to control local database.

Subpackages

• neural_compressor.ux.components.db_manager.db_models
  • neural_compressor.ux.components.db_manager.db_models.benchmark
  • neural_compressor.ux.components.db_manager.db_models.benchmark_result
  • neural_compressor.ux.components.db_manager.db_models.dataloader
  • neural_compressor.ux.components.db_manager.db_models.dataset
  • neural_compressor.ux.components.db_manager.db_models.domain
  • neural_compressor.ux.components.db_manager.db_models.domain_flavour
  • neural_compressor.ux.components.db_manager.db_models.example
  • neural_compressor.ux.components.db_manager.db_models.framework
  • neural_compressor.ux.components.db_manager.db_models.metric
  • neural_compressor.ux.components.db_manager.db_models.model
  • neural_compressor.ux.components.db_manager.db_models.optimization
  • neural_compressor.ux.components.db_manager.db_models.optimization_type
  • neural_compressor.ux.components.db_manager.db_models.precision
  • neural_compressor.ux.components.db_manager.db_models.profiling
  • neural_compressor.ux.components.db_manager.db_models.profiling_result
  • neural_compressor.ux.components.db_manager.db_models.project
  • neural_compressor.ux.components.db_manager.db_models.pruning_details
  • neural_compressor.ux.components.db_manager.db_models.transform
  • neural_compressor.ux.components.db_manager.db_models.tuning_details
  • neural_compressor.ux.components.db_manager.db_models.tuning_history
• neural_compressor.ux.components.db_manager.db_operations
  • neural_compressor.ux.components.db_manager.db_operations.benchmark_api_interface
  • neural_compressor.ux.components.db_manager.db_operations.dataset_api_interface
  • neural_compressor.ux.components.db_manager.db_operations.db_operations
  • neural_compressor.ux.components.db_manager.db_operations.diagnosis_api_interface
  • neural_compressor.ux.components.db_manager.db_operations.dictionaries_api_interface
  • neural_compressor.ux.components.db_manager.db_operations.examples_api_interface
  • neural_compressor.ux.components.db_manager.db_operations.model_api_interface
  • neural_compressor.ux.components.db_manager.db_operations.optimization_api_interface
  • neural_compressor.ux.components.db_manager.db_operations.profiling_api_interface
  • neural_compressor.ux.components.db_manager.db_operations.project_api_interface

Submodules

• neural_compressor.ux.components.db_manager.db_manager
• neural_compressor.ux.components.db_manager.params_interfaces
• neural_compressor.ux.components.db_manager.utils

Package Contents

Classes

DBManager

Database manager class.

class neural_compressor.ux.components.db_manager.DBManager(database_location: str | None = None, log_level: int | None = None)

Database manager class.

property engine: sqlalchemy.engine.Engine

Ensure that SQLAlchemy engine is created.

initialize_database() → None

Initialize database by creating engine and session.

create_sqlalchemy_engine() → sqlalchemy.engine.Engine

Create SQLAlchemy engine.

create_all() → None

Make a call to database to create all tables.