Change variable name
Modify activation function
Modify predict function
Add forward function(private)