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

This file contains the implementation of custom types and functions for tree-based machine learning models in Velox. More...

#include <fcntl.h>
#include <stdlib.h>
#include <unistd.h>
#include <cmath>
#include <iostream>
#include <memory>
#include <string>
#include "BaseFunction.h"
#include "velox/common/base/VeloxException.h"
#include "velox/common/base/tests/GTestUtils.h"
#include "velox/exec/tests/utils/AssertQueryBuilder.h"
#include "velox/exec/tests/utils/PlanBuilder.h"
#include "velox/exec/tests/utils/TempDirectoryPath.h"
#include "velox/expression/VectorFunction.h"
#include "velox/functions/Macros.h"
#include "velox/functions/Registerer.h"
#include "velox/functions/prestosql/tests/utils/FunctionBaseTest.h"
#include "velox/ml_functions/DecisionTree.h"
#include "velox/type/OpaqueCustomTypes.h"
#include "velox/vector/tests/utils/VectorTestBase.h"

Go to the source code of this file.

Classes

class  ml::TreeType
 A custom opaque type for representing tree structures. More...
 
struct  ml::TreeT
 A struct representing the custom type for trees. More...
 
class  ml::TreeTypeFactories
 Factory class for creating instances of TreeType. More...
 
class  ml::AlwaysFailingTypeFactories
 A factory class that always fails to create instances. More...
 
class  ml::VeloxTreeConstruction
 A vector function for constructing tree structures. More...
 
class  ml::VeloxTreePrediction
 A machine learning function for making predictions using tree models. More...
 
struct  ml::VeloxTreePredictionSimpleFunction< T >
 A simple function for making predictions using tree models. More...
 

Typedefs

using ml::TheTree = CustomType<TreeT>
 Alias for the custom tree type.
 

Detailed Description

This file contains the implementation of custom types and functions for tree-based machine learning models in Velox.