mirror of
https://github.com/Doctorado-ML/bayesclass.git
synced 2025-08-20 10:05:55 +00:00
20 lines
478 B
C++
20 lines
478 B
C++
#ifndef KDB_H
|
|
#define KDB_H
|
|
#include "BaseClassifier.h"
|
|
#include "utils.h"
|
|
namespace bayesnet {
|
|
using namespace std;
|
|
using namespace torch;
|
|
class KDB : public BaseClassifier {
|
|
private:
|
|
int k;
|
|
float theta;
|
|
void add_m_edges(int idx, vector<int>& S, Tensor& weights);
|
|
protected:
|
|
void train() override;
|
|
public:
|
|
KDB(int k, float theta = 0.03);
|
|
vector<string> graph(string name = "KDB") override;
|
|
};
|
|
}
|
|
#endif |