making wrapper for creating nn params and training more comprehensive