geopanag / learn_im

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Data

The graphs stem from the SNAP repository. The format of graphs is a weighted edgelist (with weighted cascade weights) in .inf, accompanied by an attribute file. The benchmark codes for influence maximization are adapted from IMM, DegreeDiscount, PMIA, FINDER and DeepIS, while for influence estimation we develop a python version of DMP inside "diffuse.py". All benchmark codes can be found in the respective folder. Unzip the data.zip in a "data" folder in the current folder.

Requirements

To run this code you will need the following in python 3.5.2:

Code

The following scripts use the default parameters mentioned in the paper.

1.Influence estimation and error using the GLIE stored model

python influence_predictions.py

2.Influence maximization (20 and 100 seeds) using the stored model with Celf-glie and evaluation of the seeds. Note that evaluation can take more then 3 hours for the large datasets.

python celf_glie.py

3.Influence maximization using the stored GNN model and the stored Grim model.

python grim.py

4.Influence maximization using the stored GNN model and the Pun model.

python pun.py

5.Train GNN on the negative samples, using the provided "influence_train_set.csv" constructed as discribed in section 4.1 of the paper.

python glie_train.py

6.Train Grim on the 50 graphs in "dql_graphs" as described in section 4.2 of the paper.

python train_dqn.py

7.The scripts in the preprocessing folder are required to create the "influence_train_set.csv".

About

License:MIT License


Languages

Language:Python 100.0%