lorenzo.volpi
/
QuaPy
forked from moreo/QuaPy
1
0
Fork 0
Code Pull Requests Activity

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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
quapy/method/neural.py
View File

@ -49,7 +49,6 @@ class QuaNetTrainer(BaseQuantifier):
} }
self.patience = patience self.patience = patience
os.makedirs(checkpointdir, exist_ok=True)
if checkpointname is None: if checkpointname is None:
local_random = random.Random() local_random = random.Random()
random_code = '-'.join(str(local_random.randint(0, 1000000)) for _ in range(5)) 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('Classifier data: ', len(classifier_data))
print('Q-Training data: ', len(train_data)) print('Q-Training data: ', len(train_data))
print('Q-Valid data: ', len(valid_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 # estimate the hard and soft stats tpr and fpr of the classifier
self.tr_prev = data.prevalence() self.tr_prev = data.prevalence()