Training on data without 'state' attribute

[theasianpianist]

If my dataset doesn't contain the state attribute, how can I still train my model? Attr.py doesn't seem to have any clues. My data doesn't have a 'states' attribute, do I need to add one and set it to blanks? Right now when I try to train the model, I get a KeyError in the data loader. Removing 'states' from the keys to use then yields an index out of range error in Attr.py.

[s-shanky-k]

@theasianpianist Were you able to find a solution this issue?

[hayaldeniz]

The index out of range error in Attr.py is most probably because one needs to adjust the embedding dimensions for the driverID's. For my case, the ID's were larger than the first parameter in embed_dims for driverID in the Attr.py. Replacing it with the maximum value of the ID for my dataset solved my problem.