search

lavis-nlp / irt

Open World KGC - IRT Dataset
MIT License
7 stars 1 forks source link

How to use IRT datasets in PyKEEN? #1

Open cthoyt opened 3 years ago

cthoyt commented 3 years ago

The readme shows an example for using PyKEEN, but not how to load one of the PyKEEN graphs with it (it looks like you need a standard pytorch dataset)

If I've got

from pykeen.datasets import Nations

dataset = Nations()

What do I do next?

cthoyt commented 3 years ago

Or maybe I misunderstood what the PyKEEN connection was - is it supposed to be that you can take the datasets from this repo and use directly in PyKEEN? If that's the case, then elaboration in the example on how to do that would be great!

kantholtz commented 3 years ago

Hi Charles - yes, it is the other way around :)

I'll leave this issue open until I refined the documentation.