adding to_DataFrame function

__init__.py

-from .toolkit import ClassificationProject
-from .compare import overlay_ROC, overlay_loss
-from .add_friend import add_friend
+from .toolkit import *
+from .compare import *
+from .add_friend import *

toolkit.py

 #!/usr/bin/env python
 
+__all__ = ["ClassificationProject"]
+
 from sys import version_info
 
 if version_info[0] > 2:
@@ -69,6 +71,7 @@ def byteify(input):
 if version_info[0] > 2:
     byteify = lambda input : input
 
+
 class ClassificationProject(object):
 
     """Simple framework to load data from ROOT TTrees and train Keras
@@ -1130,6 +1133,16 @@ class ClassificationProject(object):
         # self.plot_significance()
 
 
+    def to_DataFrame(self):
+        df = pd.DataFrame(np.concatenate([self.x_train, self.x_test]), columns=self.fields)
+        df["weight"] = np.concatenate([self.w_train, self.w_test])
+        df["labels"] = pd.Categorical.from_codes(
+            np.concatenate([self.y_train, self.y_test]),
+            categories=["background", "signal"]
+        )
+        return df
+
+
 def create_getter(dataset_name):
     def getx(self):
         if getattr(self, "_"+dataset_name) is None: