convertXGBoostToRootTree.py File Reference

Go to the source code of this file.

Classes
class	util.convertXGBoostToRootTree.XBGoostTextNode

Namespaces
namespace	util
namespace	util.convertXGBoostToRootTree

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

Variables
str	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']
	util.convertXGBoostToRootTree.output_treename = convertXGBoostToRootTree(args.input, args.output, args.tree_name)
	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)