NN and lgb baseline