Skip to content
Snippets Groups Projects

MVA Models manager

Merged MVA Models manager
dev_NeuralNetwork_framework_update_v2 into master
Merged Valerii Kholoimov requested to merge dev_NeuralNetwork_framework_update_v2 into master

Introducing MVA Models Manager, inspired by Monitoring System ( !1483 (merged)), to simplify MVA models loading:

  • MVA models are initialized inside algorithms, simplifying the model loading codes
  • Only required models in a sequence would read data from .json files
  • Provide an extendable framework to integrate gitlib2 in the future.

Was presented at 06 Aug WP2 meeting. Also adding documentation and examples of how new model can be created to Allen docs.

Example of usage:

#include "MultiLayerFCNN.cuh"
using NeuralNetworkType = Allen::MVAModels::MultiLayerNN;
NeuralNetworkType two_track_mva_nn {"two_track_mva_nn", "/allen_two_track_mva_model_June22.json"};

All existing MVA models are migrated to the framework in MR: SingleLayerFCNN (Matching Ghost Killer, Downstream Selector etc), MultiLayerFCNN (Two Track MVA, ElectronID NN), CatboostDesTree (MuonCatboostEvaluator).

FYI: @adeoyang, @dovombru, @mveghel

Edited by Valerii Kholoimov

Merge request reports

Merge request pipeline #9879076 failed

Stage: check
Stage: build
Stage: check-warnings
Stage: run
Stage: test
Stage: publish
Stage: manual trigger
Stage: build full
Stage: run full
Stage: test full
Stage: publish full
Stage: .post

Merge request pipeline failed for ef05620a

Merged by Sebastien PonceSebastien Ponce 2 months ago (Jan 14, 2025 7:40am UTC)

Loading

Pipeline #9893704 failed

Stage: check
Stage: build
Stage: check-warnings
Stage: run
Stage: test
Stage: publish
Stage: manual trigger
Stage: build full
Stage: run full
Stage: test full
Stage: publish full

Pipeline failed for 4d8759ed on master

Activity

Filter activity
  • Approvals
  • Assignees & reviewers
  • Comments (from bots)
  • Comments (from users)
  • Commits & branches
  • Edits
  • Labels
  • Lock status
  • Mentions
  • Merge request status
  • Tracking
Please register or sign in to reply