ValidatorForClassifier.h

#pragma once
 
#if _MSC_VER >= 1900 && defined(_M_X64)
 
#include "ValidatorBase.h"
#include "Tensor.h"
 
namespace FLImaging
{
    namespace AI
    {
        template <typename T>
        class FL_EXPORT CValidatorForClassifier : public CValidatorBase<T>
        {
 
        public:
            CValidatorForClassifier();
            CValidatorForClassifier(const CValidatorForClassifier<T>& validator);
            CValidatorForClassifier(const CValidatorForClassifier<T>* pValidator);
            virtual ~CValidatorForClassifier();
 
            const CValidatorForClassifier<T>& operator=(const CValidatorForClassifier<T>& validator);
 
            virtual const CResult Initialize() override;
            virtual const CResult Clear() override;
 
            virtual const CResult Assign(const CValidatorForClassifier<T>& validator);
            virtual const CResult Assign(const CValidatorForClassifier<T>* pValidator);
 
            virtual T Validate(int32_t* pI32IterationFeedback = nullptr) override;
 
            static const CResult AccumulateConfusionMatrix(const CTensor<T>* pTsrPredictMemory, const CTensor<T>* pTsrAnswerMemory, int64_t i64BatchCount, CTensor<double>* pTsrConfusionMatrix, int64_t i64ClassCount, int64_t i64IgnoreIndex = -1);
            static const CResult AccumulateConfusionMatrix(const CTensor<T>* pTsrPredictMemory, const CTensor<T>* pTsrScoreMemory, const CTensor<T>* pTsrAnswerMemory, int64_t i64BatchCount, CTensor<double>* pTsrConfusionMatrix, T tScoreThreshold, int64_t i64ClassCount, int64_t i64IgnoreIndex = -1);
 
            static const CResult IntegrationConfusionMatrix(std::vector<const CTensor<double>*>& vctCM, CTensor<double>& tsrResultCM);
            static const CResult CalculateF1(const int64_t* pI64ConfusionMatrix, T& tResultF1, int64_t i64ClassCount);
            static const CResult CalculateMacroAccuracy(const int64_t* pI64ConfusionMatrix, T& tResultF1, int64_t i64ClassCount);
 
            const CTensor<double>& GetConfusionMatrix() const;
 
            virtual const CResult SetAccuracyThreshold(T tScoreThreshold = (T)0.67);
            virtual T GetAccuracyThreshold() const;
            DeclareGetClassType();
            SupportToDuplicateObject(CValidatorForClassifier<T>, *this);
 
        protected:
 
        private:
            template<typename T>
            friend class COptimizer;
            template<typename T>
            friend class CValidatorForSemanticSegmentation;
 
        protected:
            CTensor<double> m_tsrCM;
            Base::CFLArray<Base::CFLArray<int32_t>> m_flaClassNum;
            Base::CFLArray<Base::CFLBase*> m_flaLearningIndividualAugmentationParams;
            Base::CFLArray<Base::CFLBase*> m_flaValidationIndividualAugmentationParams;
            bool m_bValidationIndividualAugmentationEnable;
            bool m_bDifferentIndividualAugmentation;
            T m_tAccuracyThreshold;
        };
 
 
        typedef CValidatorForClassifier<float> CValidatorForClassifierF;
        typedef CValidatorForClassifier<double> CValidatorForClassifierD;
    }
}
 
#endif