Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Request for the atom types labels of NCI1 datasets #50

Open
ljacg opened this issue Nov 28, 2022 · 0 comments
Open

Request for the atom types labels of NCI1 datasets #50

ljacg opened this issue Nov 28, 2022 · 0 comments

Comments

@ljacg
Copy link

ljacg commented Nov 28, 2022

I have searched almost all the literature, but have not found the atomic type corresponding to the node label of the NCI1 dataset. We know that the NCI1 dataset is made up of molecule graphs composed of atoms in 37 categories. just like the molcules in MUTAG dataset are composed of atoms in 7 categories: {0:'C', 1:'N', 2:'O', 3:'F', 4:'I', 5:'Cl', 6:'Br'}. Sincerely ask everyone to reply !!!

In ./notebooks/utils/plot_all_graphs.py, it plots graphs in MUTAG dataset with node(atom) labels like this:

# line [19 - 40]
    dataset, y = loadDataset("../../datasets/MUTAG/MUTAG_A.txt")
    for idx in [6]: #[65]:#
        G = dataset[idx]
        ncolors= []
        for node in G.nodes:
            if G.nodes[node]['atom'] == '0':
                G.nodes[node]['atom'] = 'C'
                ncolors.append('#bd3182')
            elif G.nodes[node]['atom'] == '1':
                G.nodes[node]['atom'] = 'N'
                ncolors.append('#3182bd')
            elif G.nodes[node]['atom'] == '2':
                G.nodes[node]['atom'] = 'O'
                ncolors.append('#82bd31')
            elif G.nodes[node]['atom'] == '3':
                G.nodes[node]['atom'] = 'F'
            elif G.nodes[node]['atom'] == '4':
                G.nodes[node]['atom'] = 'I'
            elif G.nodes[node]['atom'] == '5':
                G.nodes[node]['atom'] = 'Cl'
            elif G.nodes[node]['atom'] == '6':
                G.nodes[node]['atom'] = 'Br'
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

1 participant