Make pydot/graph_viz optional
Use sklearn.cross_validation for data splitting