diff --git a/lang/fr/notebooks/utils/get_dataset_attributes.py b/lang/fr/notebooks/utils/get_dataset_attributes.py new file mode 100644 index 0000000..911f573 --- /dev/null +++ b/lang/fr/notebooks/utils/get_dataset_attributes.py @@ -0,0 +1,62 @@ +#!/usr/bin/env python3 +# -*- coding: utf-8 -*- +""" +Created on Wed Oct 17 16:07:38 2018 + +@author: ljia +""" + +import sys +sys.path.insert(0, "../../") +from gklearn.utils.graphfiles import loadDataset +from gklearn.utils.graphdataset import get_dataset_attributes + +dslist = [ + {'name': 'Acyclic', 'dataset': '../../datasets/acyclic/dataset_bps.ds',}, + {'name': 'Alkane', 'dataset': '../../datasets/Alkane/dataset.ds', + 'dataset_y': '../../datasets/Alkane/dataset_boiling_point_names.txt',}, + {'name': 'MAO', 'dataset': '../../datasets/MAO/dataset.ds',}, + {'name': 'PAH', 'dataset': '../../datasets/PAH/dataset.ds',}, + {'name': 'MUTAG', 'dataset': '../../datasets/MUTAG/MUTAG.mat', + 'extra_params': {'am_sp_al_nl_el': [0, 0, 3, 1, 2]}}, + {'name': 'Letter-med', 'dataset': '../../datasets/Letter-med/Letter-med_A.txt'}, + {'name': 'ENZYMES', 'dataset': '../../datasets/ENZYMES_txt/ENZYMES_A_sparse.txt'}, + {'name': 'Mutagenicity', 'dataset': '../../datasets/Mutagenicity/Mutagenicity_A.txt'}, + {'name': 'D&D', 'dataset': '../../datasets/D&D/DD.mat', + 'extra_params': {'am_sp_al_nl_el': [0, 1, 2, 1, -1]}}, + {'name': 'AIDS', 'dataset': '../../datasets/AIDS/AIDS_A.txt'}, + {'name': 'FIRSTMM_DB', 'dataset': '../../datasets/FIRSTMM_DB/FIRSTMM_DB_A.txt'}, + {'name': 'MSRC9', 'dataset': '../../datasets/MSRC_9_txt/MSRC_9_A.txt'}, + {'name': 'MSRC21', 'dataset': '../../datasets/MSRC_21_txt/MSRC_21_A.txt'}, + {'name': 'SYNTHETIC', 'dataset': '../../datasets/SYNTHETIC_txt/SYNTHETIC_A_sparse.txt'}, + {'name': 'BZR', 'dataset': '../../datasets/BZR_txt/BZR_A_sparse.txt'}, + {'name': 'COX2', 'dataset': '../../datasets/COX2_txt/COX2_A_sparse.txt'}, + {'name': 'DHFR', 'dataset': '../../datasets/DHFR_txt/DHFR_A_sparse.txt'}, + {'name': 'PROTEINS', 'dataset': '../../datasets/PROTEINS_txt/PROTEINS_A_sparse.txt'}, + {'name': 'PROTEINS_full', 'dataset': '../../datasets/PROTEINS_full_txt/PROTEINS_full_A_sparse.txt'}, + {'name': 'NCI1', 'dataset': '../../datasets/NCI1/NCI1.mat', + 'extra_params': {'am_sp_al_nl_el': [1, 1, 2, 0, -1]}}, + {'name': 'NCI109', 'dataset': '../../datasets/NCI109/NCI109.mat', + 'extra_params': {'am_sp_al_nl_el': [1, 1, 2, 0, -1]}}, + {'name': 'NCI-HIV', 'dataset': '../../datasets/NCI-HIV/AIDO99SD.sdf', + 'dataset_y': '../../datasets/NCI-HIV/aids_conc_may04.txt',}, + +# # not working below +# {'name': 'PTC_FM', 'dataset': '../../datasets/PTC/Train/FM.ds',}, +# {'name': 'PTC_FR', 'dataset': '../../datasets/PTC/Train/FR.ds',}, +# {'name': 'PTC_MM', 'dataset': '../../datasets/PTC/Train/MM.ds',}, +# {'name': 'PTC_MR', 'dataset': '../../datasets/PTC/Train/MR.ds',}, +] + +for ds in dslist: + dataset, y = loadDataset( + ds['dataset'], + filename_y=(ds['dataset_y'] if 'dataset_y' in ds else None), + extra_params=(ds['extra_params'] if 'extra_params' in ds else None)) + attrs = get_dataset_attributes( + dataset, target=y, node_label='atom', edge_label='bond_type') + print() + print(ds['name'] + ':') + for atr in attrs: + print(atr, ':', attrs[atr]) + print() \ No newline at end of file