#pragma once

#include <vector>
#include <opencv2/ml/ml.hpp>

class RTree
{
private:
	int _size = 0;	// 数据集大小
	int _dim = 0;	// 数据集维度，属性个数

	// 测试测试：(特征向量，响应值)
	std::vector<std::pair<std::vector<double>, double>> testData;

	cv::Ptr<cv::ml::RTrees> model;
	cv::Ptr<cv::ml::TrainData> trainDataPtr = nullptr;

public:
	/*
	* @brief 创建模型
	*/
	void create();

	/*
	* @brief 从CSV中导入训练集，其实可以使用 cv::ml::loadFromCSV 函数
	* @brief 默认开头为标题，且最后一列为标签
	* @param path 数据集路径
	*/
	bool loadTrainDataFromCSV(std::string const &path);

	/*
	* @brief 从CSV中导入测试集
	* @param path 数据集路径
	*/
	bool loadTestDataFromCSV(std::string const &path);

	/*
	* @brief 训练
	*/
	void train();

	/*
	* @brief 测试
	*/
	void test();
};