22 lines
491 B
C++
22 lines
491 B
C++
#ifndef NETWORK_H
|
|
#define NETWORK_H
|
|
#include <string>
|
|
#include <vector>
|
|
#include <map>
|
|
#include "Node.h"
|
|
|
|
namespace bayesnet {
|
|
class Network {
|
|
private:
|
|
std::map<std::string, Node*> nodes;
|
|
public:
|
|
Network();
|
|
~Network();
|
|
void addNode(std::string);
|
|
void addEdge(std::string, std::string);
|
|
void fit(const std::vector<std::vector<double>>&);
|
|
std::vector<double> predict(const std::vector<std::vector<double>>&);
|
|
};
|
|
}
|
|
#endif
|