def run(self) -> Table:
with create_loading_bar(
EXPERIMENT_NAME, DATASET_COLUMN_ORDER, len(DATASET_COLUMN_ORDER)
) as d_iterable:
for dataset_name in d_iterable:
builder = DatasetBuilder(dataset_name)
builder.build()
builder.export()
print(f"{dataset_name} exported.")
return Table()
def test_export_dataset(self):
"""
Tests that after MockDataset is exported all required folders have been updated.
:return:
"""
# Setup
dataset_name = "MockDataset"
dataset_builder = DatasetBuilder(dataset_name)
dataset_builder.build()
# Work
dataset_builder.export()
folders = ["Artifacts", "Oracles"]
for folder_rel_path in folders:
path_to_folder = os.path.join(
PATH_TO_SAMPLE_DATASETS, dataset_name, folder_rel_path
)
check_folder_has_updated(path_to_folder)
from api.datasets.builder.dataset_builder import DatasetBuilder
from api.datasets.dataset import Dataset
from api.tracer import Tracer
DATASET_NAME = "IllustrativeExample"
TOP_TECHNIQUE_NAME = "(. (VSM NT) (0 1))"
BOTTOM_TECHNIQUE_NAME = "(. (VSM NT) (1 2))"
DIRECT_TECHNIQUE_NAME = "(. (VSM NT) (0 2))"
TECHNIQUE_NAME = "(x (MAX INDEPENDENT) ((. (VSM NT) (0 1)) (. (VSM NT) (1 2))))"
REBUILD = False
if __name__ == "__main__":
if REBUILD:
dataset_builder = DatasetBuilder(DATASET_NAME)
dataset_builder.build()
dataset_builder.export()
dataset = Dataset(DATASET_NAME)
tracer = Tracer()
top_technique_data = tracer.get_technique_data(DATASET_NAME,
TOP_TECHNIQUE_NAME)
bottom_technique_data = tracer.get_technique_data(DATASET_NAME,
BOTTOM_TECHNIQUE_NAME)
direct_technique_data = tracer.get_technique_data(DATASET_NAME,
DIRECT_TECHNIQUE_NAME)
top_score = top_technique_data.similarity_matrix[0][0]
bottom_score = bottom_technique_data.similarity_matrix[0][0]
transitive_score = top_score * bottom_score
direct_score = direct_technique_data.similarity_matrix[0][0]