BayesNet 1.0.7.
Bayesian Network and basic classifiers Library.
Loading...
Searching...
No Matches
bayesnet::Proposal Class Reference
Inheritance diagram for bayesnet::Proposal:

Public Member Functions

 Proposal (torch::Tensor &pDataset, std::vector< std::string > &features_, std::string &className_)
 

Protected Member Functions

void checkInput (const torch::Tensor &X, const torch::Tensor &y)
 
torch::Tensor prepareX (torch::Tensor &X)
 
map< std::string, std::vector< int > > localDiscretizationProposal (const map< std::string, std::vector< int > > &states, Network &model)
 
map< std::string, std::vector< int > > fit_local_discretization (const torch::Tensor &y)
 

Protected Attributes

torch::Tensor Xf
 
torch::Tensor y
 
map< std::string, mdlp::CPPFImdlp * > discretizers
 

Detailed Description

Definition at line 17 of file Proposal.h.

Constructor & Destructor Documentation

◆ Proposal()

bayesnet::Proposal::Proposal ( torch::Tensor & pDataset,
std::vector< std::string > & features_,
std::string & className_ )

Definition at line 10 of file Proposal.cc.

◆ ~Proposal()

bayesnet::Proposal::~Proposal ( )
virtual

Definition at line 11 of file Proposal.cc.

Member Function Documentation

◆ checkInput()

void bayesnet::Proposal::checkInput ( const torch::Tensor & X,
const torch::Tensor & y )
protected

Definition at line 17 of file Proposal.cc.

◆ fit_local_discretization()

map< std::string, std::vector< int > > bayesnet::Proposal::fit_local_discretization ( const torch::Tensor & y)
protected

Definition at line 77 of file Proposal.cc.

◆ localDiscretizationProposal()

map< std::string, std::vector< int > > bayesnet::Proposal::localDiscretizationProposal ( const map< std::string, std::vector< int > > & states,
Network & model )
protected

Definition at line 26 of file Proposal.cc.

◆ prepareX()

torch::Tensor bayesnet::Proposal::prepareX ( torch::Tensor & X)
protected

Definition at line 104 of file Proposal.cc.

Member Data Documentation

◆ discretizers

map<std::string, mdlp::CPPFImdlp*> bayesnet::Proposal::discretizers
protected

Definition at line 28 of file Proposal.h.

◆ Xf

torch::Tensor bayesnet::Proposal::Xf
protected

Definition at line 26 of file Proposal.h.

◆ y

torch::Tensor bayesnet::Proposal::y
protected

Definition at line 27 of file Proposal.h.


The documentation for this class was generated from the following files: