Result.h

Go to the documentation of this file.

#ifndef LCVN_RESULT_H
#define LCVN_RESULT_H

#include "larrecodnn/CVN/func/InteractionType.h"
#include <math.h>
#include <vector>

namespace lcvn {
  class Result {
  public:
    Result(const float* output, unsigned int& nOutputs);
    // Vector version of the constructor
    Result(const std::vector<std::vector<float>> output);
    //Result(const float* output, unsigned int& nOutputs, const float* features, unsigned int& nFeatures);
    Result();

    unsigned int ArgMax(int output_n) const;

    float Max();

    TFIsAntineutrino PredictedIsAntineutrino() const;

    TFFlavour PredictedFlavour() const;

    TFInteraction PredictedInteraction() const;

    TFTopologyProtons PredictedProtons() const;

    TFTopologyPions PredictedPions() const;

    TFTopologyPizeros PredictedPizeros() const;

    TFTopologyNeutrons PredictedNeutrons() const;

    float GetIsAntineutrinoProbability() const;

    float GetNumuProbability() const;

    float GetNueProbability() const;

    float GetNutauProbability() const;

    float GetNCProbability() const;

    float GetQEProbability() const;

    float GetResProbability() const;

    float GetDISProbability() const;

    float GetOtherProbability() const;

    float Get0protonsProbability() const;

    float Get1protonsProbability() const;

    float Get2protonsProbability() const;

    float GetNprotonsProbability() const;

    float Get0pionsProbability() const;

    float Get1pionsProbability() const;

    float Get2pionsProbability() const;

    float GetNpionsProbability() const;

    float Get0pizerosProbability() const;

    float Get1pizerosProbability() const;

    float Get2pizerosProbability() const;

    float GetNpizerosProbability() const;

    float Get0neutronsProbability() const;

    float Get1neutronsProbability() const;

    float Get2neutronsProbability() const;

    float GetNneutronsProbability() const;

    unsigned int NOutput();

    std::vector<std::vector<float>> fOutput; 
  };
}

#endif // CVN_RESULT_H