softmax test module that uses sklearn optimizer