Hugging Face's logo

Spaces:
keras-io
/
TF-GB-Forest
Runtime error

App Files Community
tdubon commited on
Commit
bd62f8b
·
1 Parent(s): 4f62f60

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -1
app.py CHANGED
@@ -59,7 +59,6 @@ test_ds = tfdf.keras.pd_dataframe_to_tf_dataset(test_data, label="income_level")
59
  # Train a GB Trees model
60
  model = tfdf.keras.GradientBoostedTreesModel(
61
  features = [feature_a, feature_b, feature_c, feature_d],
62
- exclude_non_specified_features = True,
63
  growing_strategy = "BEST_FIRST_GLOBAL",
64
  num_trees = 350,
65
  max_depth = 7,
 
59
  # Train a GB Trees model
60
  model = tfdf.keras.GradientBoostedTreesModel(
61
  features = [feature_a, feature_b, feature_c, feature_d],
 
62
  growing_strategy = "BEST_FIRST_GLOBAL",
63
  num_trees = 350,
64
  max_depth = 7,