Issue with prediciton

[f-izzat]

Followed the same procedure as in CATCHM_demo.ipynb with my own dataset. However got the error

"""
Traceback (most recent call last):
  File "/Users/faiz/miniconda3/envs/ccfraud/lib/python3.7/multiprocessing/pool.py", line 121, in worker
    result = (True, func(*args, **kwds))
  File "/Users/faiz/Desktop/work/code/credit_card_fraud/catchm.py", line 130, in inductive_pooling_chunk
    pooled_embedding = get_pooled_embedding(receiver_neighbors, embeddings, gamma)
  File "/Users/faiz/Desktop/work/code/credit_card_fraud/catchm.py", line 143, in get_pooled_embedding
    embeddings_to_pool = embeddings[neighbors, :]
IndexError: index 268273 is out of bounds for axis 0 with size 10285
"""

Format for X_train, X_test, y_train, y_test remains the same as in the notebook. Running with the transactions.txt dataset works fine. I wonder what is required to make my dataset work?

FYI: I put all functions and classes into a single catchm.py file.