convertXGBoostToRootTree.py File Reference

Go to the source code of this file.

Classes
class	util.convertXGBoostToRootTree.XBGoostTextNode

Namespaces
	util.convertXGBoostToRootTree

Functions
def	util.convertXGBoostToRootTree.dump_tree (tree_structure)
def	util.convertXGBoostToRootTree.dump2ROOT (model, output_filename, output_treename='xgboost')
def	util.convertXGBoostToRootTree.convertXGBoostToRootTree (model, output_filename, tree_name='xgboost')
def	util.convertXGBoostToRootTree.test (model_file, tree_file, objective, tree_name='xgboost', ntests=10000, test_file=None)
def	util.convertXGBoostToRootTree.test_regression (booster, mva_utils, objective, ntests=10000, test_file=None)
def	util.convertXGBoostToRootTree.test_binary (booster, mva_utils, objective, ntests=10000, test_file=None)
def	util.convertXGBoostToRootTree.test_multiclass (booster, mva_utils, objective, ntests=10000, test_file=None)
def	util.convertXGBoostToRootTree.check_file (fn)

Variables
string	util.convertXGBoostToRootTree.__doc__ = "Convert XGBoost model to TTree to be used with MVAUtils."
string	util.convertXGBoostToRootTree.__author__ = "Yuan-Tang Chou"
	util.convertXGBoostToRootTree.level
	util.convertXGBoostToRootTree.parser = argparse.ArgumentParser(description=__doc__)
	util.convertXGBoostToRootTree.help
	util.convertXGBoostToRootTree.type
	util.convertXGBoostToRootTree.str
	util.convertXGBoostToRootTree.default
	util.convertXGBoostToRootTree.action
	util.convertXGBoostToRootTree.int
	util.convertXGBoostToRootTree.args = parser.parse_args()
list	util.convertXGBoostToRootTree.supported_objective = ['binary:logistic', 'reg:linear', 'reg:squarederror','multi:softprob']
def	util.convertXGBoostToRootTree.output_treename = convertXGBoostToRootTree(args.input, args.output, args.tree_name)
def	util.convertXGBoostToRootTree.result = test(args.input, args.output, args.objective, args.tree_name, args.ntests, args.test_file)
	util.convertXGBoostToRootTree.objective = args.objective
	util.convertXGBoostToRootTree.data = np.load(args.test_file)