ENH: Add visualization to python package

This commit is contained in:
sinhrks
2015-08-11 16:40:09 +09:00
parent a7202ee804
commit d24b36adf9
9 changed files with 311 additions and 2 deletions

View File

@@ -8,9 +8,11 @@ from __future__ import absolute_import
from .core import DMatrix, Booster
from .training import train, cv
from .sklearn import XGBModel, XGBClassifier, XGBRegressor
from .plotting import plot_importance, plot_tree, to_graphviz
__version__ = '0.4'
__all__ = ['DMatrix', 'Booster',
'train', 'cv',
'XGBModel', 'XGBClassifier', 'XGBRegressor']
'XGBModel', 'XGBClassifier', 'XGBRegressor',
'plot_importance', 'plot_tree', 'to_graphviz']