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]