26#ifdef USE_INFERENCE_LWTNN
29#include <lwtnn/parse_json.hh>
34Par04LwtnnInference::Par04LwtnnInference(
G4String modelPath)
38 std::ifstream input(modelPath);
40 fGraph = std::make_unique<lwt::LightweightGraph>(lwt::parse_json_graph(input));
46void Par04LwtnnInference::RunInference(std::vector<float> aGenVector,
47 std::vector<G4double>& aEnergies,
51 fNetworkInputs inputs;
52 for(std::size_t i = 0; i < aGenVector.size(); ++i)
54 inputs[
"node_0"][
"variable_" + std::to_string(i)] = aGenVector[i];
58 fNetworkOutputs outputs = fGraph->compute(inputs);
59 aEnergies.assign(aSize, 0);
60 for(
int i = 0; i < aSize; i++)
61 aEnergies[i] = outputs[
"out_" + std::to_string(i)];