ML functions
 
Loading...
Searching...
No Matches
DecisionTree.h File Reference

Implementation of a decision tree for machine learning predictions. More...

#include <fcntl.h>
#include <stdlib.h>
#include <unistd.h>
#include <cmath>
#include <iostream>
#include <memory>
#include <string>
#include "velox/exec/tests/utils/AssertQueryBuilder.h"
#include "velox/exec/tests/utils/PlanBuilder.h"
#include "velox/exec/tests/utils/TempDirectoryPath.h"
#include "velox/ml_functions/BaseFunction.h"
#include "velox/vector/tests/utils/VectorTestBase.h"

Go to the source code of this file.

Classes

struct  ml::Node
 Represents a node in a decision tree. More...
 
class  ml::Tree
 Represents a decision tree used for predictions. More...
 
class  ml::TreePrediction
 Implements a machine learning function for tree-based predictions. More...
 

Macros

#define MAX_NUM_NODES_PER_TREE   512
 

Typedefs

typedef std::shared_ptr< Treeml::TreePtr
 

Detailed Description

Implementation of a decision tree for machine learning predictions.