From 72aeb7c1b8645f9f938aa61ec938fb35a8bff247 Mon Sep 17 00:00:00 2001
From: Nikolai <osterei33@gmx.de>
Date: Fri, 27 Apr 2018 10:10:52 +0200
Subject: [PATCH] make method naming consistent (snake case)
---
toolkit.py | 8 ++++----
1 file changed, 4 insertions(+), 4 deletions(-)
diff --git a/toolkit.py b/toolkit.py
index 9159901..9955e0d 100755
--- a/toolkit.py
+++ b/toolkit.py
@@ -305,7 +305,7 @@ class KerasROOTClassification:
def evaluate(self):
pass
- def writeFriendTree(self):
+ def write_friend_tree(self):
pass
@@ -354,7 +354,7 @@ class KerasROOTClassification:
fig.savefig(os.path.join(plot_dir, "var_{}.pdf".format(var_index)))
- def plotROC(self):
+ def plot_ROC(self):
logger.info("Plot ROC curve")
fpr, tpr, threshold = roc_curve(self.y_test, self.scores_test, sample_weight = self.w_test)
@@ -375,7 +375,7 @@ class KerasROOTClassification:
plt.clf()
- def plotScore(self):
+ def plot_score(self):
pass
@@ -398,4 +398,4 @@ if __name__ == "__main__":
identifiers = ["DatasetNumber", "EventNumber"])
c.train(epochs=1)
- c.plotROC()
+ c.plot_ROC()
--
GitLab