Add option to choose activation function during nplm training