def eval_baseline(weight, behavior):
opt = Options().parse()
opt.baseline = True
opt.sequence_length = 20
opt.behavior_layer = 0
opt.data_dir = '../data/' + behavior
print("Model Config: ", opt)
model = Model(opt)
model.load_weight(weight)
return model.evaluate(0, keep_frame=True)
def eval_proposed(weight, use_haptic, use_audio, use_virbo, behavior):
opt = Options().parse()
opt.use_behavior = True
opt.use_haptic = use_haptic
opt.use_audio = use_audio
opt.use_vibro = False
opt.behavior_layer = 1
opt.aux = True
opt.sequence_length = 20
opt.data_dir = '../data/' + behavior
print("Model Config: ", opt)
model = Model(opt)
model.load_weight(weight)
return model.evaluate(0, keep_frame=True)
