hex.tree.xgboost

Class XGBoostUtils.FeatureProperties

java.lang.Object

hex.tree.xgboost.XGBoostUtils.FeatureProperties

Enclosing class:

XGBoostUtils

public static class XGBoostUtils.FeatureProperties
extends java.lang.Object

Field Summary

Fields

Modifier and Type	Field and Description
java.lang.String[]	_names
boolean[]	_oneHotEncoded
int[]	_originalColumnIndices
java.lang.String[]	_originalNames
java.util.Map<java.lang.String,java.lang.Integer>	_originalNamesMap

Constructor Summary

Constructors

Constructor and Description
FeatureProperties(java.lang.String[] originalNames, java.lang.String[] names, boolean[] oneHotEncoded, int[] originalColumnIndices)

Method Summary

Modifier and Type	Method and Description
int	getOriginalIndex(java.lang.String originalName)
java.lang.Integer[]	mapOriginalNamesToIndices(java.lang.String[] names)

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Field Detail

_originalNames

public java.lang.String[] _originalNames

_originalNamesMap

public java.util.Map<java.lang.String,java.lang.Integer> _originalNamesMap

_names

public java.lang.String[] _names

_oneHotEncoded

public boolean[] _oneHotEncoded

_originalColumnIndices

public int[] _originalColumnIndices

Constructor Detail

FeatureProperties

public FeatureProperties(java.lang.String[] originalNames,
                         java.lang.String[] names,
                         boolean[] oneHotEncoded,
                         int[] originalColumnIndices)

Method Detail

getOriginalIndex

public int getOriginalIndex(java.lang.String originalName)

mapOriginalNamesToIndices

public java.lang.Integer[] mapOriginalNamesToIndices(java.lang.String[] names)