bugfinder.base.dataset

Base classes for managing dataset

class bugfinder.base.dataset.CodeWeaknessClassificationDataset(dataset_path, silent=False)

Bases: object

Main dataset class.

append_summary(op_call, op_category, exec_time, op_stats, return_code=-1)

Append new processing to summary file

clear_queue()

Clear processing queue

get_classes()

List classes identified in the dataset and their id

get_features_info()

Retrieve feature information

ignored_dirs = ['joern.db', 'neo4j_v3.db', 'features', 'models', 'embeddings']

load_summary()

Load summary file

process(silent=False)

Run all processing in the processing queue

queue_operation(op_class, op_args=None)

Queue operation

rebuild_index()

Rebuild index

reset_summary()

Reset summary file

save_summary()

Save summary file

class bugfinder.base.dataset.DatasetQueueRetCode(value)

Bases: IntEnum

Enumeration to determine the state of the processing queue.

EMPTY_QUEUE = 1

INVALID_QUEUE = 2

OK = 0

OPERATION_FAIL = 3