Now, ONN only works for classification problems as you can see in the example. ONN can be used as well for regression problems but you will need to change the loss function in the code. For ONN_THS I really don't know if it will converge using regression loss (it could stay very unstable), but you can give a try :D
Now, ONN only works for classification problems as you can see in the example. ONN can be used as well for regression problems but you will need to change the loss function in the code. For ONN_THS I really don't know if it will converge using regression loss (it could stay very unstable), but you can give a try :D