fix choosing the activation func

[SimonSuster]

In predict(), output obtained with tanh can never be returned due to an error in the conditional statement. This fix makes it possible to choose any of tanh, relu or other.