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