def __init__(self, n_classes, savepath, train_metrics, valid_metrics,
best_metric, best_type, verbose=False, *args):
super(Callback, self).__init__()
# Save input parameters
self.n_classes = n_classes
self.savepath = savepath
self.verbose = verbose
self.train_metrics = train_metrics
self.valid_metrics = valid_metrics
self.best_metric = best_metric
self.best_type = best_type
评论列表
文章目录