* Fix input validation * Fix output_dim not being set if features_only=True and output_dim_added_layer not None