From 54dc2980e6f50138212f58c648b7922a9054a6ac Mon Sep 17 00:00:00 2001 From: Alex Moreo Date: Fri, 22 Jan 2021 10:03:08 +0100 Subject: [PATCH] creating checkpointdir in fit --- quapy/method/neural.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/quapy/method/neural.py b/quapy/method/neural.py index 1a030f3..6ff9e28 100644 --- a/quapy/method/neural.py +++ b/quapy/method/neural.py @@ -49,7 +49,6 @@ class QuaNetTrainer(BaseQuantifier): } self.patience = patience - os.makedirs(checkpointdir, exist_ok=True) if checkpointname is None: local_random = random.Random() random_code = '-'.join(str(local_random.randint(0, 1000000)) for _ in range(5)) @@ -74,6 +73,7 @@ class QuaNetTrainer(BaseQuantifier): print('Classifier data: ', len(classifier_data)) print('Q-Training data: ', len(train_data)) print('Q-Valid data: ', len(valid_data)) + os.makedirs(self.checkpointdir, exist_ok=True) # estimate the hard and soft stats tpr and fpr of the classifier self.tr_prev = data.prevalence()