fix poisson evaluator to use the correct poisson likelihood

heal-research · Apr 5, 2024 · cda6146 · cda6146
1 parent 7e485b4
commit cda6146
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/include/pyoperon/optimizer.hpp b/include/pyoperon/optimizer.hpp
@@ -12,14 +12,16 @@ using TEvaluator               = Operon::Evaluator<TDispatch>;
 using TMDLEvaluator            = Operon::MinimumDescriptionLengthEvaluator<TDispatch>;
 using TBICEvaluator            = Operon::BayesianInformationCriterionEvaluator<TDispatch>;
 using TAIKEvaluator            = Operon::AkaikeInformationCriterionEvaluator<TDispatch>;
-using TGaussEvaluator          = Operon::GaussianLikelihoodEvaluator<TDispatch>;
-using TPoissonEvaluator        = Operon::PoissonLikelihoodEvaluator<TDispatch>;
 
 // likelihood
 using TGaussianLikelihood      = Operon::GaussianLikelihood<Operon::Scalar>;
 using TPoissonLikelihood       = Operon::PoissonLikelihood<Operon::Scalar, false>;
 using TPoissonLikelihoodLog    = Operon::PoissonLikelihood<Operon::Scalar, true>;
 
+// likelihood evaluators
+using TGaussEvaluator          = Operon::GaussianLikelihoodEvaluator<TDispatch>;
+using TPoissonEvaluator        = Operon::LikelihoodEvaluator<TDispatch, TPoissonLikelihood>;
+using TPoissonLogEvaluator     = Operon::LikelihoodEvaluator<TDispatch, TPoissonLikelihoodLog>;
 // optimizer
 using TOptimizerBase           = Operon::OptimizerBase<TDispatch>;
 
@@ -84,4 +86,4 @@ class Optimizer {
 
     [[nodiscard]] auto Get() const { return optimizer_.get(); }
 };
-}  // namespace detail
+}  // namespace detail