Hugging Face's logo

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

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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -56,9 +56,11 @@ feature_d = tfdf.keras.FeatureUsage(name="capital_losses", semantic=tfdf.keras.F
56
  train_ds = tfdf.keras.pd_dataframe_to_tf_dataset(train_data, label="income_level")
57
  test_ds = tfdf.keras.pd_dataframe_to_tf_dataset(test_data, label="income_level")
58
 
 
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,
 
56
  train_ds = tfdf.keras.pd_dataframe_to_tf_dataset(train_data, label="income_level")
57
  test_ds = tfdf.keras.pd_dataframe_to_tf_dataset(test_data, label="income_level")
58
 
59
+ tf.autograph.set_verbosity(0)
60
  # Train a GB Trees model
61
  model = tfdf.keras.GradientBoostedTreesModel(
62
  features = [feature_a, feature_b, feature_c, feature_d],
63
+ exclude_non_specified_features = True,
64
  growing_strategy = "BEST_FIRST_GLOBAL",
65
  num_trees = 350,
66
  max_depth = 7,