Angel Martinez

[BUG] Target in schema causes incorrect down scoring of false negative.

3

### Bug description Target in schema causes incorrect down scoring of false negative. With target in the schema, the positive item ids are all made the values of the target...

[BUG] Batch Size results in different prediction outputs.

1

### Bug description Predictions from different size data loader yields different results. Reproducing: 1. simplistic two tower model ``` model = mm.TwoTowerModel( schema, query_tower=mm.MLPBlock([64,64], no_activation_last_layer=True, ), item_tower=mm.MLPBlock([64,64], no_activation_last_layer=True, )) model.compile(...

[BUG] partition_on throws an error if not a list.

1

**Describe the bug** Small bug, but when saving a parquet file and using partition_on I believe it looks for index 0 regardless of its input, so a text saying partition_on...

Running the example on the read me returns a 2,32 embedding. You get the correct result if you change to call function to the following (removing the unsqueeze): ```def __call__(self,...

Default Time Emb Dim is 6, however, the paper states that the value used was 12. The example training commands don't specify this hyper parameter. To recreate the results in...