How can I use the trained model for prediction? is it right to use prediction_model.predict(new_x) ?

yaringal / multi-task-learning-example

A multi-task learning example for the paper https://arxiv.org/abs/1705.07115

MIT License

838 stars 205 forks source link

How can I use the trained model for prediction? is it right to use prediction_model.predict(new_x) ? #17

Open Qianqian-Yang opened 2 years ago

Qianqian-Yang commented 2 years ago

I'm not in the field of deep learning and computer science, but I found this work very interesting. I am confused about what should I do if I want to use the trained model for prediction? Can I achieve this goal through prediction_model.predict(new_x)? I see only the trainable_model was trained but it can not achieve prediction. Has the prediction_model been trained at the same time? Thanks very much.

Yhyyy-hhu commented 1 year ago

YES, the only change with the new method is on the loss function

yyruuu commented 1 year ago

The value I predicted with prediction_model is very different from the real value. Do you have any similar problems?