def save(self, location="brain"):
""" Pickle the brain """
if self._trained:
joblib.dump(self.lobe, location + ".pickle")
logger.info('Brain %s saved', location + '.pickle')
else:
return logger.error('Brain is not trained yet! Nothing to save...')
评论列表
文章目录