Skip to content
Neural Consciousness Flow
Python Shell
Branch: master
Clone or download
Fetching latest commit…
Cannot retrieve the latest commit at this time.
Permalink
Type Name Latest commit message Commit time
Failed to load latest commit information.
code
data
output/NELL995_subgraph
visual/NELL995_subgraph
LICENSE.txt
README.md
requirements.txt
run.sh

README.md

NeuCFlow

Neural Consciousness Flow

Training and Evaluating

./run.sh --dataset <Dataset>

can be one of 'FB237', 'FB237_v2', 'FB15K', 'WN18RR', 'WN18RR_v2', 'WN', 'YAGO310', 'NELL995'.

Visualization

Run with test_output_attention to get data files of extracted subgraphs for each query in test. For example, if you want to get data files on the NELL995 dataset (containing several separate datasets), with max_attended_nodes=20, run:

./run.sh --dataset NELL995 --test_output_attention --max_attended_nodes 20 --test_max_attended_nodes 20

Then, you get data files in the output/NELL995_subgraph directory. Next, visualize them by:

cd code
python visualize.py --dataset NELL995

You will find image files for visualization in the visual/NELL995_subgraph directory.

You can’t perform that action at this time.