colda.database.test_database package

Submodules

colda.database.test_database.algorithm_database module

class colda.database.test_database.algorithm_database.TestAlgorithmDatabase

Bases: BaseDatabase, AbstractAlgorithmDatabase

Store and manage test_algorithm database. Test algorithm database mainly stores the output and log generated from algorithm part of test stage.

Methods

get_instance get_all_records store_record get_record

classmethod delete()

get_all_records_history() → list[tuple[str, str]]

Return all keys of records in this database

Returns

list[tuple[str, str]]

classmethod get_instance() → type[colda.database.test_database.algorithm_database.TestAlgorithmDatabase]

Singleton pattern. Get instance of current class.

Returns

type[TestAlgorithmDatabase]

get_record(user_id: str, test_id: str, algorithm_data_name: str) → None

Get record

Parameters

user_id : str test_id : str algorithm_data_name : str

Returns

None

store_record(user_id: str, test_id: str, algorithm_data_name: str, algorithm_data: list[str] | list[list[str]] | Any) → None

Store record

Parameters

user_id : str test_id : str algorithm_data_name : str algorithm_data : Union[list[str], list[list[str]], Any]

Returns

None

colda.database.test_database.api module

class colda.database.test_database.api.TestAlgorithmDatabase

Bases: BaseDatabase, AbstractAlgorithmDatabase

Store and manage test_algorithm database. Test algorithm database mainly stores the output and log generated from algorithm part of test stage.

Methods

get_instance get_all_records store_record get_record

classmethod delete()

get_all_records_history() → list[tuple[str, str]]

Return all keys of records in this database

Returns

list[tuple[str, str]]

classmethod get_instance() → type[database.test_database.algorithm_database.TestAlgorithmDatabase]

Singleton pattern. Get instance of current class.

Returns

type[TestAlgorithmDatabase]

get_record(user_id: str, test_id: str, algorithm_data_name: str) → None

Get record

Parameters

user_id : str test_id : str algorithm_data_name : str

Returns

None

store_record(user_id: str, test_id: str, algorithm_data_name: str, algorithm_data: list[str] | list[list[str]] | Any) → None

Store record

Parameters

user_id : str test_id : str algorithm_data_name : str algorithm_data : Union[list[str], list[list[str]], Any]

Returns

None

class colda.database.test_database.api.TestAssistorMetadataDatabase

Bases: BaseDatabase, AbstractMetadataDatabase

Store and manage data generated from assistor test stage.

Methods

get_instance get_all_records store_record get_record

classmethod delete()

get_all_records_history() → list[tuple[str, str]]

Return all keys of records in this database

Returns

list[tuple[str, str]]

classmethod get_instance() → type[database.test_database.assistor_metadata_database.TestAssistorMetadataDatabase]

Singleton pattern. Get instance of current class.

Returns

type[TestAssistorMetadataDatabase]

get_record(user_id: str, test_id: str) → None

Get record

Parameters

user_id : str test_id : str

Returns

None

store_record(user_id: str, train_id: str, mode: str, task_mode: Literal['classification', 'regression'], model_name: Literal['linear', 'decision_tree', 'svm', 'gradient_boosting', 'mlp'], test_id: str, test_file_path: str, test_id_column: str, test_data_column: str, test_name: str | None = None, test_description: str | None = None) → None

Store record

Parameters

user_id : str train_id : str mode : str task_mode : Task_Mode model_name : Model_Name test_id : str test_file_path : str test_id_column : str test_data_column : str test_name : str=None test_description : str=None

Returns

None

class colda.database.test_database.api.TestSponsorMetadataDatabase

Bases: BaseDatabase, AbstractMetadataDatabase

Store and manage data generated from sponsor test stage.

Methods

get_instance get_all_records store_record get_record

classmethod delete()

get_all_records_history() → list[tuple[str, str]]

Return all keys of records in this database

Returns

list[tuple[str, str]]

classmethod get_instance() → type[database.test_database.sponsor_metadata_database.TestSponsorMetadataDatabase]

Singleton pattern. Get instance of current class.

Returns

type[TestAssistorMetadataDatabase]

get_record(user_id: str, test_id: str) → None

Get record

Parameters

user_id : str test_id : str

Returns

None

store_record(user_id: str, train_id: str, task_mode: Literal['classification', 'regression'], model_name: Literal['linear', 'decision_tree', 'svm', 'gradient_boosting', 'mlp'], metric_name: Literal['MAD', 'RMSE', 'R2', 'Accuracy', 'F1', 'AUCROC'], test_id: str, test_file_path: str, test_id_column: str, test_data_column: str, test_target_column: str, test_name: str | None = None, test_description: str | None = None) → None

Store record

Parameters

user_id : str train_id : str task_mode : Task_Mode model_name : Model_Name metric_name : Metric_Name test_id : str test_file_path : str test_id_column : str test_data_column : str test_target_column : str test_name : str=None test_description : str=None

Returns

None

colda.database.test_database.assistor_metadata_database module

class colda.database.test_database.assistor_metadata_database.TestAssistorMetadataDatabase

Bases: BaseDatabase, AbstractMetadataDatabase

Store and manage data generated from assistor test stage.

Methods

get_instance get_all_records store_record get_record

classmethod delete()

get_all_records_history() → list[tuple[str, str]]

Return all keys of records in this database

Returns

list[tuple[str, str]]

classmethod get_instance() → type[colda.database.test_database.assistor_metadata_database.TestAssistorMetadataDatabase]

Singleton pattern. Get instance of current class.

Returns

type[TestAssistorMetadataDatabase]

get_record(user_id: str, test_id: str) → None

Get record

Parameters

user_id : str test_id : str

Returns

None

store_record(user_id: str, train_id: str, mode: str, task_mode: Literal['classification', 'regression'], model_name: Literal['linear', 'decision_tree', 'svm', 'gradient_boosting', 'mlp'], test_id: str, test_file_path: str, test_id_column: str, test_data_column: str, test_name: str | None = None, test_description: str | None = None) → None

Store record

Parameters

user_id : str train_id : str mode : str task_mode : Task_Mode model_name : Model_Name test_id : str test_file_path : str test_id_column : str test_data_column : str test_name : str=None test_description : str=None

Returns

None

colda.database.test_database.sponsor_metadata_database module

class colda.database.test_database.sponsor_metadata_database.TestSponsorMetadataDatabase

Bases: BaseDatabase, AbstractMetadataDatabase

Store and manage data generated from sponsor test stage.

Methods

get_instance get_all_records store_record get_record

classmethod delete()

get_all_records_history() → list[tuple[str, str]]

Return all keys of records in this database

Returns

list[tuple[str, str]]

classmethod get_instance() → type[colda.database.test_database.sponsor_metadata_database.TestSponsorMetadataDatabase]

Singleton pattern. Get instance of current class.

Returns

type[TestAssistorMetadataDatabase]

get_record(user_id: str, test_id: str) → None

Get record

Parameters

user_id : str test_id : str

Returns

None

store_record(user_id: str, train_id: str, task_mode: Literal['classification', 'regression'], model_name: Literal['linear', 'decision_tree', 'svm', 'gradient_boosting', 'mlp'], metric_name: Literal['MAD', 'RMSE', 'R2', 'Accuracy', 'F1', 'AUCROC'], test_id: str, test_file_path: str, test_id_column: str, test_data_column: str, test_target_column: str, test_name: str | None = None, test_description: str | None = None) → None

Store record

Parameters

user_id : str train_id : str task_mode : Task_Mode model_name : Model_Name metric_name : Metric_Name test_id : str test_file_path : str test_id_column : str test_data_column : str test_target_column : str test_name : str=None test_description : str=None

Returns

None

Module contents