baseline NN implementation