139 lines
3.7 KiB
C
139 lines
3.7 KiB
C
|
#pragma once
|
|||
|
|
/**
|
|||
|
|
* @file dist.h
|
|||
|
|
* @brief 确定分布类型
|
|||
|
|
* 根据输入的数据特征确定分布类型
|
|||
|
|
* 并给出对应的置信区间
|
|||
|
|
* @author your name (you@domain.com)
|
|||
|
|
* @version 0.1
|
|||
|
|
* @date 2024-02-21
|
|||
|
|
*
|
|||
|
|
* Copyright: Baosight Co. Ltd.
|
|||
|
|
* DO NOT COPY/USE WITHOUT PERMISSION
|
|||
|
|
*
|
|||
|
|
*/
|
|||
|
|
#include <dlib/statistics.h>
|
|||
|
|
#include <log4cplus/LOG.h>
|
|||
|
|
#include <algorithm>
|
|||
|
|
#include <boost/math/distributions/chi_squared.hpp>
|
|||
|
|
#include <boost/math/distributions/normal.hpp>
|
|||
|
|
#include <boost/math/distributions/skew_normal.hpp>
|
|||
|
|
#include <boost/math/distributions/students_t.hpp>
|
|||
|
|
#include <memory>
|
|||
|
|
#include <tuple>
|
|||
|
|
#include <vector>
|
|||
|
|
#include "mix_cc/type/range.h"
|
|||
|
|
|
|||
|
|
namespace DAA {
|
|||
|
|
|
|||
|
|
using boost::math::quantile;
|
|||
|
|
/**
|
|||
|
|
* @brief 分布验证和置信区间获取
|
|||
|
|
* 先根据Prob确定出错的最大可能性
|
|||
|
|
* 再根据最大可能性、数据分布区间，得到详细的数据作用范围
|
|||
|
|
* 根据数据的真实分布情况，重新调整参数，直到报警率小于目标数值
|
|||
|
|
* 目前只支持高斯分布类型的分布
|
|||
|
|
*/
|
|||
|
|
class Dist {
|
|||
|
|
public:
|
|||
|
|
enum class DistTypes {
|
|||
|
|
unknown,
|
|||
|
|
student_t, ///< T分布
|
|||
|
|
normal, ///<正态分布
|
|||
|
|
skew_normal ///<偏态分布
|
|||
|
|
}; ///< 预设三种分布
|
|||
|
|
constexpr static double test_prec = 0.01;
|
|||
|
|
|
|||
|
|
protected:
|
|||
|
|
DistTypes dist_type_; ///< 分布类型
|
|||
|
|
bool valid_; ///< 可用性
|
|||
|
|
dlib::running_stats<double> rs_; ///< dlib 数据分布特征
|
|||
|
|
mix_cc::float_range_t legal_range_; ///< 合法分布区间
|
|||
|
|
std::string rule_id_;
|
|||
|
|
std::shared_ptr<boost::math::normal> normal_; ///< 正态分布
|
|||
|
|
std::shared_ptr<boost::math::students_t> students_t_; ///< T分布
|
|||
|
|
std::shared_ptr<boost::math::skew_normal> skew_normal_; ///< 偏态分布
|
|||
|
|
std::unique_ptr<LOG> logger_;
|
|||
|
|
double predefied_prob_ = 0.975; ///< 预定义的置信概率 3σ的大概置信度
|
|||
|
|
|
|||
|
|
double prob_ = 0.5;
|
|||
|
|
|
|||
|
|
public:
|
|||
|
|
/**
|
|||
|
|
* @brief 构建分布检测
|
|||
|
|
*/
|
|||
|
|
Dist(/* args */);
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* @brief Destroy the Dist object
|
|||
|
|
*/
|
|||
|
|
~Dist();
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* @brief 得到分布类型
|
|||
|
|
* @return DistTypes
|
|||
|
|
*/
|
|||
|
|
DistTypes get_distribution_type() const;
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* @brief 设置置信概率
|
|||
|
|
* @param prob 置信概率
|
|||
|
|
* @return int
|
|||
|
|
*/
|
|||
|
|
int set_predefined_prob(double prob);
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* @brief Get the shifted prob object
|
|||
|
|
* @return double
|
|||
|
|
*/
|
|||
|
|
double get_shifted_prob() const;
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* @brief
|
|||
|
|
* 测试合法数据的分布类型和理想的置信区局,该函数只会根据合法数据进行判断
|
|||
|
|
* @param rs dlib::running_stats
|
|||
|
|
* @param tmp_data My Param doc
|
|||
|
|
* @return int
|
|||
|
|
*/
|
|||
|
|
int auto_test(dlib::running_stats<double> rs,
|
|||
|
|
const std::vector<double>& tmp_data);
|
|||
|
|
|
|||
|
|
// /**
|
|||
|
|
// * @brief 设置报警样本（可选，用来验证第二类错误）
|
|||
|
|
// * @param warning_sample My Param doc
|
|||
|
|
// * @return int
|
|||
|
|
// */
|
|||
|
|
// int set_warning_sample_optional(const std::vector<double>& warning_sample);
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* @brief 分布是否具有合理类型
|
|||
|
|
* @return true
|
|||
|
|
* @return false
|
|||
|
|
*/
|
|||
|
|
bool valid() const;
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* @brief 得到合法范围大小
|
|||
|
|
* @return mix_cc::float_range_t
|
|||
|
|
*/
|
|||
|
|
mix_cc::float_range_t get_range() const;
|
|||
|
|
|
|||
|
|
protected:
|
|||
|
|
/**
|
|||
|
|
* @brief 得到第一类错误的概率
|
|||
|
|
* @param prob 置信概率
|
|||
|
|
* @param tmp_data 用来验证的临时数据
|
|||
|
|
* @return double
|
|||
|
|
*/
|
|||
|
|
double get_error_rate_type_1(const double& prob,
|
|||
|
|
const std::vector<double>& tmp_data);
|
|||
|
|
/**
|
|||
|
|
* @brief 得到指定置信概率的区间
|
|||
|
|
* @param prob 置信概率
|
|||
|
|
* @return mix_cc::float_range_t
|
|||
|
|
*/
|
|||
|
|
mix_cc::float_range_t get_range(double prob);
|
|||
|
|
};
|
|||
|
|
|
|||
|
|
} // namespace DAA
|