rmontanana/BayesNet
1
0
Fork 0
Code Issues 2 Pull Requests 1 Actions Packages Projects Releases 6 Wiki Activity

Begin experiment

Browse Source
This commit is contained in:
Ricardo Montañana Gómez
2023-07-23 01:47:57 +02:00
parent 9981ad1811
commit 644b6c9be0
11 changed files with 107 additions and 118 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
src/Platform/platformUtils.h
View File

@@ -1,5 +1,6 @@
#ifndef PLATFORM_UTILS_H
#define PLATFORM_UTILS_H
#include <torch/torch.h>
#include <string>
#include <vector>
#include <map>
@@ -12,4 +13,6 @@ const string PATH = "../../data/";
bool file_exists(const std::string& name);
pair<vector<mdlp::labels_t>, map<string, int>> discretize(vector<mdlp::samples_t>& X, mdlp::labels_t& y, vector<string> features);
tuple<vector<vector<int>>, vector<int>, vector<string>, string, map<string, vector<int>>> loadFile(string name);
tuple<torch::Tensor, torch::Tensor, vector<string>> loadDataset(string name, bool discretize);
pair <map<string, int>, map<string, vector<int>>> discretize_info(torch::Tensor& X, torch::Tensor& y);
#endif //PLATFORM_UTILS_H