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. | |
This file contains the implementation of custom types and functions for tree-based machine learning models in Velox.