search

zjukg / AdaMF-MAT

[Paper][LREC-COLING 2024] Unleashing the Power of Imbalanced Modality Information for Multi-modal Knowledge Graph Completion
22 stars 3 forks source link

pytorch version of testdataloader #5

Closed LittleMonster104 closed 5 months ago

lizilong777 commented 5 months ago

Can you provide the pytorch version of testdataloader? I didn't find this version. Hope to get your reply.

Zhang-Each commented 5 months ago

Hello, the test dataloader class is in the path mmkgc/data/TestDataLoader.py

lizilong777 commented 5 months ago

Hello, the test dataloader class is in the path mmkgc/data/TestDataLoader.py

Hello, the test dataloader class is in the path mmkgc/data/TestDataLoader.py

Thank you for your reply, but the version I want doesn't use Base.so. I am looking for something like mmkgc/data/PyTorchTrainDataLoader. Do you have this version of TestDataLoader?

Zhang-Each commented 5 months ago

You can refer to the OpenKE project (https://github.com/thunlp/OpenKE/) and find the C++ files in openke/base. Our Base.so is compiled from these C++ files.

Zhang-Each commented 5 months ago

Our code is totally based on OpenKE. Hence, if you want to modify the source code, you should refer to OpenKE.

lizilong777 commented 5 months ago

Our code is totally based on OpenKE. Hence, if you want to modify the source code, you should refer to OpenKE.

Thank you very much for your reply.