test_result_base.hpp

#pragma once
 
#include "../_cexports/c_polimago.h"
 
#include "../global.hpp"
#include "../string.hpp"
#include "predictor_base.hpp"
 
#include <vector>
 
namespace Cvb
{
CVB_BEGIN_INLINE_NS
 
namespace Polimago
{
namespace Testing
{
 
 
class TestResultBase : public PolimagoFactoryCreatedObject
{
protected:
  TestResultBase (ReleaseObjectGuard&& guard)
    : PolimagoFactoryCreatedObject (std::move (guard)),
      exampleMeanSquareErrors_()
  {
    CVB_CALL_CAPI(PMGetTestResultTrainParams(Handle(), trainingParameters_));
 
    auto num_examples = NumExamples ();
    for (decltype(num_examples) i = 0; i < num_examples; ++i)
    {
      exampleMeanSquareErrors_.push_back (CVB_CALL_CAPI (PMGetTestResultExampleMeanSquareError(Handle(), i)));
    }
  }
 
  TestResultBase (TestResultBase&&) noexcept = default;
 
  TestResultBase& operator=(TestResultBase&&) noexcept = default;
 
  virtual ~TestResultBase () = default;
 
  // Helper to load a saved predictor from file
  static CExports::TTESTRESULT LoadInternal (const String & fileName)
  {
    CExports::TTESTRESULT testResult = nullptr;
 
    CVB_CALL_CAPI_CHECKED (PMOpenTestResultTyped(fileName.c_str(), testResult));
    return testResult;
  }
 
  void SaveFunction (const String &fileName) const override
  {
    CVB_CALL_CAPI_CHECKED (PMSaveTestResultTyped(fileName.c_str(), Handle()));
  }
 
public:
 
  int NumExamples() const
  {
    return static_cast<int>(CVB_CALL_CAPI (PMGetTestResultNumExamples(Handle())));
  }
 
 
  double TotalMeanSquareError() const
  {
    return static_cast<int>(CVB_CALL_CAPI (PMGetTestResultTotalMeanSquareError(Handle())));
  }
 
 
  std::vector<double> ExampleMeanSquareError() const
  {
    return exampleMeanSquareErrors_;
  }
 
 
  int FeatureResolution() const noexcept
  {
    return trainingParameters_.FeatureMap.FeatureResolution;
  }
 
private:
  std::vector<double> exampleMeanSquareErrors_;
};
 
using TestResultBasePtr = std::shared_ptr<TestResultBase>;
 
 
} /* namespace Testing */
 
using Testing::TestResultBase;
using Testing::TestResultBasePtr;
 
} /* namespace Polimago */
CVB_END_INLINE_NS
} /* namespace Cvb */