- [ ] Replace the [Factorization](https://github.com/kilian-group/banditeval/blob/ffc29e6cdbf9cf170bae534933e48ad47b733edc/banditeval/factorization.py#L5) method that they use with scalar/vanilla NN from #2 - [ ] Compare the performance of this method versus original method