Arguments
This function internally uses scikit-learn function DecisionTreeRegressor through teradataml Open source ML functions.
Transformed data won't have a features column.
| PySpark Argument Name | Open Source Function Argument Name | Notes |
|---|---|---|
| predictionCol | Not yet available. | |
| varianceCol | Not yet available. | |
| maxDepth | max_depth | |
| maxBins | Not yet available. | |
| minInstancesPerNode | min_samples_split | PySpark takes int value (>= 1) default value is 1, teradatamlspk supports int value (>1) or float in (0.0, 1.0]. Default value is 1.0. |
| minInfoGain | Not yet available. | |
| maxMemoryInMB | Not yet available. | |
| cacheNodeIds | Not yet available. | |
| checkpointInterval | Not yet available. | |
| impurity | Not yet available. | Default value is squared_error, supported by osml, PySpark variance is not yet supported. |
| seed | random_state | |
| weightCol | Not yet available. | |
| leafCol | Not yet available. | |
| minWeightFractionPerNode | min_weight_fraction_leaf |
Attributes/Methods
| Attribute/Method Name | Supported | Notes |
|---|---|---|
| clear | ||
| copy | ||
| explainParam | ||
| explainParams | ||
| extractParamMap | ||
| fit | ||
| fitMultiple | ||
| getCacheNodeIds | ||
| getCheckpointInterval | ||
| getFeaturesCol | ||
| getImpurity | ||
| getLabelCol | ||
| getLeafCol | ||
| getMaxBins | ||
| getMaxDepth | ||
| getMaxMemoryInMB | ||
| getMinInfoGain | ||
| getMinInstancesPerNode | ||
| getMinWeightFractionPerNode | ||
| getOrDefault | ||
| getParam | ||
| getPredictionCol | ||
| getSeed | ||
| getVarianceCol | ||
| getWeightCol | ||
| hasDefault | ||
| hasParam | ||
| isDefined | ||
| isSet | ||
| load | ||
| read | ||
| save | ||
| set | ||
| setCacheNodeIds | ||
| setCheckpointInterval | ||
| setFeaturesCol | ||
| setFitIntercept | ||
| setImpurity | ||
| setLabelCol | ||
| setLeafCol | ||
| setMaxBins | ||
| setMaxDepth | ||
| setMaxMemoryInMB | ||
| setMinInfoGain | ||
| setMinWeightFractionPerNode | ||
| setParams | ||
| setPredictionCol | ||
| setSeed | ||
| setVarianceCol | ||
| setWeightCol | ||
| write | ||
| cacheNodeIds | ||
| checkpointInterval | ||
| featuresCol | ||
| impurity | ||
| labelCol | ||
| leafCol | ||
| maxBins | ||
| maxDepth | ||
| maxMemoryInMB | ||
| minInfoGain | ||
| minInstancesPerNode | ||
| minWeightFractionPerNode | ||
| params | ||
| predictionCol | ||
| seed | ||
| supportedImpurities | ||
| varianceCol | ||
| weightCol |