Skip to content

Draft: Fixing gaussian regression

Mathias Josef Backes requested to merge backes/salt:main into main

Some minor code changes in order to fix Gaussian regression. Details about the changes:

  • predictionwriter.py L.162: attached some zeros in order not to run into shape-related errors
  • task.py L.164: applied a softplus function and the necessary scaling in order to obtain the correct values for sigma (compare L.207)
  • task.py L.209: reshaped mean and sigmas to be in proper format for the loss function (main bug)

In general it is necessary that the Gaussian regression tasks are named "gaussian_regression" in order for the changes to work.

Merge request reports