Adding neural network inference to NMMA

[malinadesai]

A new function has been added to /nmma/em/ananlysis.py called nnanalysis that can accept either a light curve or an injection file and will produce a corner plot displaying the inference results. Helper functions and model weights are added to nmma/mlmodel. Running this will require the following two commands to be added when creating the nmma environment: conda install pytorch torchvision -c pytorch
pip install nflows

[tsunhopang]

@malinadesai could you update the requirement to include the new packages required?

[sahiljhawar]

@malinadesai can you also add a test for neuralnet. See here

nmma/nmma/tests/analysis.py

Line 93 in a83be30

args.interpolation_type = "sklearn_gp"

Also, could you add the conda command (and any pitfalls that user might encounter) in the documentation (Installation page) and also a brief intro on how to use the neuralnet. I suggest to open an new PR for the documentation.

[sahiljhawar]

Can you also add the input files required, I can try adding the test.

[malinadesai]

@tsunhopang @sahiljhawar I added an ml_requirements.txt file that contains torch and nflows, and i was able to successfully create a new environment when running pip install -r ml_requirements.txt. I also added lfi_analysis.md to the doc directory with example use and addressing the limitations. I also added a new test function to nmma/tests/analysis.py called def test_nn_analysis(args). I'm not sure if that function is the appropriate way to add a test, so it could definitely benefit from a review.

[sahiljhawar]

hehe, thanks for the test

[sahiljhawar]

@mcoughlin @tsunhopang is there a need of refactoring since nnanalysis have a lot of similar code to analysis?

[mcoughlin]

@sahiljhawar @tsunhopang I suggest prioritizing getting this merged, and work on a refactor in a second PR.

[sahiljhawar]

Tests are passing, LGTM

[mcoughlin]

@sahiljhawar feel free to give an approving review and merge!

[sahiljhawar]

@malinadesai you should reset your main branch with NMMA main branch. As @mcoughlin once said, "Please use a branch for next time."