XGBoostRegressor¶
Input¶
It takes in a DataFrame as input and performs XGBoost Regression
Output¶
The XGBoost Model generated is passed along to the next nodes. The input DataFrame is also passed along to the next nodes
Type¶
ml-estimator
Class¶
fire.nodes.ml.NodeXGBoostRegressor
Fields¶
| Name | Title | Description |
|---|---|---|
| featuresCol | Features Column | Features column of type vectorUDT for model fitting |
| labelCol | Label Column | The label column for model fitting |
| predictionCol | Prediction Column | The prediction column created during model scoring. |
| maxDepth | Max Depth | The Maximum depth of a tree |
| maxBins | Max Bins | The maximum number of bins used for discretizing continuous features.Must be >= 2 and >= number of categories in any categorical feature. |
| maxLeaves | Max Leaves | |
| numRound | Num Round | |
| numWorkers | Num Workers | |
| objective | Objective | |
| eta | Eta | |
| regLambda | Reg Lambda | |
| regAlpha | Reg Alpha | |
| subsample | Subsample | |
| sampleType | SampleType | |
| treeMethod | TreeMethod | |
| useExternalMemory | UseExternalMemory | |
| seed | Seed | |
| baseScore | Base Score | |
| minChildWeight | Min Child Weight | |
| colsampleBylevel | ColSampleByLevel | |
| colsampleBytree | ColSampleByTree | |
| minSplitLoss | MinSplitLoss | |
| maxDeltaStep | MaxDeltaStep | |
| sketchEps | SketchEps | |
| scalePosWeight | ScalePosWeight | |
| growPlicy | GrowPlicy | |
| normalizeType | NormalizeType | |
| skipDrop | SkipDrop | |
| rateDrop | RateDrop |