DBN regression - Githubissues

Jxinyee / dbn_pytorch

dbn with smote 增强

11 stars 3 forks source link

Open JIAZhenZ opened 3 years ago

JIAZhenZ commented 3 years ago

I wonder to know that how should I revise the code if I want to solve a regression question by DBN? Looking forward for your answer, tanks.

Jxinyee commented 2 years ago

you can rewrite the softmax activate function. use a full-connected node to replace it and 'mse' is the loss function