[BUG] Error while retraining

[gabryarri]

I tried to retrain the model using new tags but this happened:

line 16, in nll_loss
loss += criterion(pred[i], ground_truth[i])
IndexError: index 12 is out of bounds for dimension 0 with size 12

[David Beauchemin]

That looks like empty tags list for some addresses.

Try installing the dev version with the following command pip install -U git+https://github.com/GRAAL-Research/deepparse.git@dev. I've added some data tests (e.g. no empty tags list for an address) and improved error handling for such cases.

[David Beauchemin]

See the new release for better error handling.

[gabryarri]

I have successfully retrained the model but when I try to use it to do new predictions the program gives me this error:

AttributeError: 'NoneType' object has no attribute 'lower'

This is the code I wrote:

address_parser = AddressParser(path_to_retrained_model=my_path)

I also tried to specify the model type but nothing changes

[David Beauchemin]

It seems like some addresses are None value in your dataset. We apply a cleaning process onto the addresses (coma cleaning and lower cleaning).

Will improve the validation of dataset with this case also.