Neural Subgraph Isomorphism Counting实现记录
发布时间
阅读量:
阅读量
train.py文件的输入参数配置如下:
--model
RGIN
--predict_net
DIAMNet
--predict_net_mem_init
mean
--predict_net_mem_len
4
--predict_net_recurrent_steps
3
--gpu_id
-1
--batch_size
512
--max_npv
8
--max_npe
8
--max_npvl
8
--max_npel
8
--max_ngv
64
--max_nge
256
--max_ngvl
16
--max_ngel
16
--pattern_dir
../data/debug/patterns
--graph_dir
../data/debug/graphs
--metadata_dir
../data/debug/metadata
--save_data_dir
../data/debug
--save_model_dir
../dum
全部评论 (0)
还没有任何评论哟~
