I am having difficulty which format my KG data should be in to be used with pykeen. Wh

Hi <a class="user-mention notranslate" data-hovercard-type="user" data-hovercard-url="

Data format description about pykeen HOT 2 CLOSED

pykeen commented on May 25, 2024

Data format description

from pykeen.

Comments (2)

cthoyt commented on May 25, 2024

Hi @ChristopherMarais, the format is quite simple - you can have a TSV file with three columns corresponding to the head, relation, and tail. Each row should be a single triple.

The tutorial at https://pykeen.readthedocs.io/en/latest/tutorial/byod.html should help you get started with using your own data. If you've got a specific uncertainty with this, I'd be glad to walk you through it and make any updates that will help you and other users!

Happy Friday, greets from Germany

from pykeen.

ChristopherMarais commented on May 25, 2024

Hi thank you for the quick response. I managed to figure out the exact format by looking at the incorporated datasets. however I run into an error when using:

tf = TriplesFactory(path=work_path + '/data.txt')
training, testing = tf.split()

I started a new issue with this error as the main topic, I'm not sure why it would give that error so I flagged it as a bug.

The error is the following:

ValueError                                Traceback (most recent call last)
<ipython-input-15-ef15ccdc9011> in <module>
      3 
      4 tf = TriplesFactory(path=work_path + '/eucalyptus_triplets.txt')
----> 5 training, testing = tf.split()
      6 
      7 pipeline_result = pipeline(

~\anaconda3\envs\pykeen\lib\site-packages\pykeen\triples\triples_factory.py in split(self, ratios, random_state, randomize_cleanup)
    419         idx = np.arange(n_triples)
    420         if random_state is None:
--> 421             random_state = np.random.randint(0, 2 ** 32 - 1, dtype=np.int64)
    422             logger.warning(f'Using random_state={random_state} to split {self}')
    423         if isinstance(random_state, int):

mtrand.pyx in numpy.random.mtrand.RandomState.randint()

_bounded_integers.pyx in numpy.random._bounded_integers._rand_int32()

ValueError: high is out of bounds for int32