Added support for nn (single layer) in sklearn interface