def add_accuracy(self, model, data_loader, classifier=None, name='train'):
models = [VAEC, SVAEC]
if type(model) in models or classifier:
accuracy = compute_accuracy(model, data_loader, classifier)
self.history["Accuracy_%s" % name].append(accuracy)
if self.verbose:
print("Accuracy %s is: %4f" % (name, accuracy))
def accuracy(self, name, verbose=False):
model, cls = (self.sampling_model, self.model) if hasattr(
self, 'sampling_model') else (self.model, None)
acc = compute_accuracy(model, self.data_loaders[name], classifier=cls)
if verbose:
print("Acc for %s is : %.4f" % (name, acc))
return acc
def accuracy(self, name, verbose=False):
acc = compute_accuracy(self.model,
self.data_loaders[name],
use_cuda=self.use_cuda)
if verbose:
print("Acc for %s is : %.4f" % (name, acc))
return acc