Skip to content
This repository has been archived by the owner on Aug 23, 2022. It is now read-only.

How to deal with regression task? #8

Closed
Answerfxy opened this issue Jul 13, 2021 · 2 comments
Closed

How to deal with regression task? #8

Answerfxy opened this issue Jul 13, 2021 · 2 comments

Comments

@Answerfxy
Copy link

Answerfxy commented Jul 13, 2021

Hi Khirotaka,

Thanks for proving the code.
I am a little confused about how to convert the default classification task to the regression one.

As you said in the README file, there are two steps:
Specify the number of output dimensions in num_class.
Inherit class SAnD and overwrite ClassificationModule with RegressionModule.

Am I did correct in the following steps:

  1. Cuz it's a regression problem, num_class should be set to 1.
  2. Also, I just simply convert ClassificationModule to RegressionModule in the model.py file.

Thanks for your time!

@khirotaka
Copy link
Owner

There is no problem with the procedure you showed. As a side note, if the output is set to 1, the information one step ahead should be predicted, so if you want to predict future information (for example, 100 steps ahead), you can set num_class = 100.

However, SAnD should have been proposed as a classification model, so you can't tell if regression works until you try it.

@Answerfxy
Copy link
Author

There is no problem with the procedure you showed. As a side note, if the output is set to 1, the information one step ahead should be predicted, so if you want to predict future information (for example, 100 steps ahead), you can set num_class = 100.

However, SAnD should have been proposed as a classification model, so you can't tell if regression works until you try it.

Thanks for the explanation about num_class in regression!

Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants