1
0
Fork 0

creating checkpointdir in fit

This commit is contained in:
Alejandro Moreo Fernandez 2021-01-22 10:03:08 +01:00
parent bf1cc74ba1
commit 54dc2980e6
1 changed files with 1 additions and 1 deletions

View File

@ -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()