The function is a simple MLP that takes in a numerical input and outputs a scalar value.
