search

JayYip / m3tl

BERT for Multitask Learning
https://jayyip.github.io/m3tl/
Apache License 2.0
545 stars 125 forks source link

NameError: name 'decode_predictions' is not defined #103

Closed hoangthangta closed 2 years ago

hoangthangta commented 2 years ago

I had this error when inferring the inputs: NameError: name 'decode_predictions' is not defined. 

pred = predict_bert_multitask(
        problem = problem,
        inputs = inputs, 
        model_dir = model.params.ckpt_dir,
        problem_type_dict = problem_type_dict,
        processing_fn_dict = processing_fn_dict, 
        decode_prediction = True
        #return_model = True
        )

How to fix, thank!

JayYip commented 2 years ago

This is a bug. Thanks for reporting.