AssertionError: Unknown dataset three_all

[Su-hong-yu]

`suhongyu@su:~/my_task$ dglke_train --model_name TransE_l2 --data_path /home/suhongyu/my_task/data/three_all --dataset three_all --data_files train.txt valid.txt test.txt \

Traceback (most recent call last): File "/home/suhongyu/anaconda3/envs/openke/bin/dglke_train", line 8, in sys.exit(main()) File "/home/suhongyu/anaconda3/envs/openke/lib/python3.8/site-packages/dglke/train.py", line 81, in main dataset = get_dataset(args.data_path, File "/home/suhongyu/anaconda3/envs/openke/lib/python3.8/site-packages/dglke/dataloader/KGDataset.py", line 644, in get_dataset assert False, "Unknown dataset {}".format(data_name) AssertionError: Unknown dataset three_all `

what shoud I do to solve this problem?

[zahramahani]

hi i found problem according to your address of error you need to define format for your data so add this: --format raw_udd_htr or --format udd_htr

to learn more check documention: dgl_doc

@Su-hong-yu