Geant4 Cross Reference

Cross-Referencing   Geant4
Geant4/examples/extended/parameterisations/Par04/include/Par04LwtnnInference.hh

Version: [ ReleaseNotes ] [ 1.0 ] [ 1.1 ] [ 2.0 ] [ 3.0 ] [ 3.1 ] [ 3.2 ] [ 4.0 ] [ 4.0.p1 ] [ 4.0.p2 ] [ 4.1 ] [ 4.1.p1 ] [ 5.0 ] [ 5.0.p1 ] [ 5.1 ] [ 5.1.p1 ] [ 5.2 ] [ 5.2.p1 ] [ 5.2.p2 ] [ 6.0 ] [ 6.0.p1 ] [ 6.1 ] [ 6.2 ] [ 6.2.p1 ] [ 6.2.p2 ] [ 7.0 ] [ 7.0.p1 ] [ 7.1 ] [ 7.1.p1 ] [ 8.0 ] [ 8.0.p1 ] [ 8.1 ] [ 8.1.p1 ] [ 8.1.p2 ] [ 8.2 ] [ 8.2.p1 ] [ 8.3 ] [ 8.3.p1 ] [ 8.3.p2 ] [ 9.0 ] [ 9.0.p1 ] [ 9.0.p2 ] [ 9.1 ] [ 9.1.p1 ] [ 9.1.p2 ] [ 9.1.p3 ] [ 9.2 ] [ 9.2.p1 ] [ 9.2.p2 ] [ 9.2.p3 ] [ 9.2.p4 ] [ 9.3 ] [ 9.3.p1 ] [ 9.3.p2 ] [ 9.4 ] [ 9.4.p1 ] [ 9.4.p2 ] [ 9.4.p3 ] [ 9.4.p4 ] [ 9.5 ] [ 9.5.p1 ] [ 9.5.p2 ] [ 9.6 ] [ 9.6.p1 ] [ 9.6.p2 ] [ 9.6.p3 ] [ 9.6.p4 ] [ 10.0 ] [ 10.0.p1 ] [ 10.0.p2 ] [ 10.0.p3 ] [ 10.0.p4 ] [ 10.1 ] [ 10.1.p1 ] [ 10.1.p2 ] [ 10.1.p3 ] [ 10.2 ] [ 10.2.p1 ] [ 10.2.p2 ] [ 10.2.p3 ] [ 10.3 ] [ 10.3.p1 ] [ 10.3.p2 ] [ 10.3.p3 ] [ 10.4 ] [ 10.4.p1 ] [ 10.4.p2 ] [ 10.4.p3 ] [ 10.5 ] [ 10.5.p1 ] [ 10.6 ] [ 10.6.p1 ] [ 10.6.p2 ] [ 10.6.p3 ] [ 10.7 ] [ 10.7.p1 ] [ 10.7.p2 ] [ 10.7.p3 ] [ 10.7.p4 ] [ 11.0 ] [ 11.0.p1 ] [ 11.0.p2 ] [ 11.0.p3, ] [ 11.0.p4 ] [ 11.1 ] [ 11.1.1 ] [ 11.1.2 ] [ 11.1.3 ] [ 11.2 ] [ 11.2.1 ] [ 11.2.2 ] [ 11.3.0 ]

Diff markup

Differences between /examples/extended/parameterisations/Par04/include/Par04LwtnnInference.hh (Version 11.3.0) and /examples/extended/parameterisations/Par04/include/Par04LwtnnInference.hh (Version 11.0.p1)


  1 //                                                  1 //
  2 // *******************************************      2 // ********************************************************************
  3 // * License and Disclaimer                         3 // * License and Disclaimer                                           *
  4 // *                                                4 // *                                                                  *
  5 // * The  Geant4 software  is  copyright of th      5 // * The  Geant4 software  is  copyright of the Copyright Holders  of *
  6 // * the Geant4 Collaboration.  It is provided      6 // * the Geant4 Collaboration.  It is provided  under  the terms  and *
  7 // * conditions of the Geant4 Software License      7 // * conditions of the Geant4 Software License,  included in the file *
  8 // * LICENSE and available at  http://cern.ch/      8 // * LICENSE and available at  http://cern.ch/geant4/license .  These *
  9 // * include a list of copyright holders.           9 // * include a list of copyright holders.                             *
 10 // *                                               10 // *                                                                  *
 11 // * Neither the authors of this software syst     11 // * Neither the authors of this software system, nor their employing *
 12 // * institutes,nor the agencies providing fin     12 // * institutes,nor the agencies providing financial support for this *
 13 // * work  make  any representation or  warran     13 // * work  make  any representation or  warranty, express or implied, *
 14 // * regarding  this  software system or assum     14 // * regarding  this  software system or assume any liability for its *
 15 // * use.  Please see the license in the file      15 // * use.  Please see the license in the file  LICENSE  and URL above *
 16 // * for the full disclaimer and the limitatio     16 // * for the full disclaimer and the limitation of liability.         *
 17 // *                                               17 // *                                                                  *
 18 // * This  code  implementation is the result      18 // * This  code  implementation is the result of  the  scientific and *
 19 // * technical work of the GEANT4 collaboratio     19 // * technical work of the GEANT4 collaboration.                      *
 20 // * By using,  copying,  modifying or  distri     20 // * By using,  copying,  modifying or  distributing the software (or *
 21 // * any work based  on the software)  you  ag     21 // * any work based  on the software)  you  agree  to acknowledge its *
 22 // * use  in  resulting  scientific  publicati     22 // * use  in  resulting  scientific  publications,  and indicate your *
 23 // * acceptance of all terms of the Geant4 Sof     23 // * acceptance of all terms of the Geant4 Software license.          *
 24 // *******************************************     24 // ********************************************************************
 25 //                                                 25 //
 26                                                    26 
 27 #ifdef USE_INFERENCE_LWTNN                         27 #ifdef USE_INFERENCE_LWTNN
 28 #  ifndef PAR04LWTNNINFERENCE_HH               <<  28 #ifndef PAR04LWTNNINFERENCE_HH
 29 #    define PAR04LWTNNINFERENCE_HH             <<  29 #define PAR04LWTNNINFERENCE_HH
 30                                                    30 
 31 #    include "Par04InferenceInterface.hh"  //  <<  31 #include "Par04InferenceInterface.hh"
 32 #    include "lwtnn/LightweightGraph.hh"  // f <<  32 #include ""
 33                                                <<  33 #include ""
 34 #    include <G4String.hh>  // for G4String    <<  34 #include <fstream>
 35 #    include <G4Types.hh>  // for G4double     << 
 36 #    include <map>  // for map, map<>::value_c << 
 37 #    include <memory>  // for unique_ptr       << 
 38 #    include <string>  // for string, basic_st << 
 39 #    include <vector>  // for vector           << 
 40                                                    35 
 41 /**                                                36 /**
 42  * @brief Inference using the LWTNN library.       37  * @brief Inference using the LWTNN library.
 43  *                                                 38  *
 44  * Initializes a computation object or a light     39  * Initializes a computation object or a light weight graph using the ML model saved
 45  * as a JSON file.                                 40  * as a JSON file.
 46  * Runs the inference by evaluating the graph      41  * Runs the inference by evaluating the graph on the input vector from Par04InferenceSetup.
 47  *                                                 42  *
 48  **/                                               43  **/
 49                                                    44 
                                                   >>  45 using namespace std;
                                                   >>  46 
 50 class Par04LwtnnInference : public Par04Infere     47 class Par04LwtnnInference : public Par04InferenceInterface
 51 {                                                  48 {
 52   public:                                      <<  49  public:
 53     Par04LwtnnInference(G4String);             <<  50   Par04LwtnnInference(G4String);
 54     Par04LwtnnInference();                     <<  51   Par04LwtnnInference();
 55                                                <<  52 
 56     /// Run inference                          <<  53   /// Run inference
 57     /// @param[in] aGenVector Input latent spa <<  54   /// @param[in] aGenVector Input latent space and conditions
 58     /// @param[out] aEnergies Model output = g <<  55   /// @param[out] aEnergies Model output = generated shower energies
 59     /// @param[in] aSize Size of the output    <<  56   /// @param[in] aSize Size of the output
 60     void RunInference(std::vector<float> aGenV <<  57   void RunInference(vector<float> aGenVector, std::vector<G4double>& aEnergies, int aSize);
 61                                                <<  58 
 62   private:                                     <<  59  private:
 63     /// LWTNN graph , Network input and output <<  60   /// LWTNN graph , Network input and output vectors
 64     std::unique_ptr<lwt::LightweightGraph> fGr <<  61   std::unique_ptr<lwt::LightweightGraph> fGraph;
 65     typedef std::map<std::string, std::map<std <<  62   typedef std::map<std::string, std::map<std::string, double>> fNetworkInputs;
 66     typedef std::map<std::string, double> fNet <<  63   typedef std::map<std::string, double> fNetworkOutputs;
 67 };                                                 64 };
 68                                                    65 
 69 #  endif /* PAR04LWTNNINFERENCE_HH */          <<  66 #endif /* PAR04LWTNNINFERENCE_HH */
 70 #endif                                             67 #endif
 71                                                    68