Skip to content

Update tests for Nan Regression

Nikita Ivvan Pond requested to merge npond/salt:update_tests into main

Fixes some issues in !141 (merged).

  • Adds nan_regression unit test to gitlab cli.
  • Fixes regression task in cases where some labels are NaN. Previous approach seems not to work, due to some weird inner workings of pytorch.
Edited by Nikita Ivvan Pond

Merge request reports