mirror of
https://github.com/rmontanana/mdlp.git
synced 2025-08-20 09:55:56 +00:00
41 lines
1.0 KiB
C++
41 lines
1.0 KiB
C++
// ****************************************************************
|
|
// SPDX - FileCopyrightText: Copyright 2024 Ricardo Montañana Gómez
|
|
// SPDX - FileType: SOURCE
|
|
// SPDX - License - Identifier: MIT
|
|
// ****************************************************************
|
|
|
|
#ifndef ARFFFILES_H
|
|
#define ARFFFILES_H
|
|
|
|
#include <string>
|
|
#include <vector>
|
|
#include "../typesFImdlp.h"
|
|
|
|
using namespace std;
|
|
|
|
class ArffFiles {
|
|
private:
|
|
vector<string> lines;
|
|
vector<pair<string, string>> attributes;
|
|
string className;
|
|
string classType;
|
|
vector<mdlp::samples_t> X;
|
|
vector<int> y;
|
|
|
|
void generateDataset(bool);
|
|
|
|
public:
|
|
ArffFiles();
|
|
void load(const string&, bool = true);
|
|
vector<string> getLines() const;
|
|
unsigned long int getSize() const;
|
|
string getClassName() const;
|
|
string getClassType() const;
|
|
static string trim(const string&);
|
|
vector<mdlp::samples_t>& getX();
|
|
vector<int>& getY();
|
|
vector<pair<string, string>> getAttributes() const;
|
|
static vector<int> factorize(const vector<string>& labels_t);
|
|
};
|
|
|
|
#endif |