RandomForestClassifier Class Reference

Implements a Random Forest Classifier. More...

#include <RandomForestClassifier.hpp>

Public Member Functions
	RandomForestClassifier (int n_estimators=10, int max_depth=5, int min_samples_split=2, int max_features=-1)
	Constructs a RandomForestClassifier. More...
	~RandomForestClassifier ()=default
	Destructor for RandomForestClassifier.
void	fit (const std::vector< std::vector< double >> &X, const std::vector< int > &y)
	Fits the model to the training data. More...
std::vector< int >	predict (const std::vector< std::vector< double >> &X) const
	Predicts class labels for given input data. More...

Detailed Description

Implements a Random Forest Classifier.

Constructor & Destructor Documentation

RandomForestClassifier()

RandomForestClassifier::RandomForestClassifier	(	int	n_estimators = 10,
		int	max_depth = 5,
		int	min_samples_split = 2,
		int	max_features = -1
	)

Constructs a RandomForestClassifier.

Parameters

n_estimators	The number of trees in the forest.
max_depth	The maximum depth of the tree.
min_samples_split	The minimum number of samples required to split an internal node.
max_features	The number of features to consider when looking for the best split. Defaults to sqrt(num_features).

Member Function Documentation

fit()

void RandomForestClassifier::fit	(	const std::vector< std::vector< double >> &	X,
		const std::vector< int > &	y
	)

Fits the model to the training data.

Parameters

X	A vector of feature vectors.
y	A vector of target class labels.

predict()

std::vector< int > RandomForestClassifier::predict

(

const std::vector< std::vector< double >> &

X

)

const

Predicts class labels for given input data.

Parameters

X	A vector of feature vectors.

Returns

A vector of predicted class labels.

---

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

• ml_library_include/ml/tree/RandomForestClassifier.hpp