Building a TFBT model for exoplanet detection

In this section, we shall build the gradient boosted trees model for detecting exoplanets using the Kepler dataset. Let us follow these steps in the Jupyter Notebook to build and train the exoplanet finder model:

  1. We will save the names of all the features in a vector with the following code:
numeric_column_headers = x_train.columns.values.tolist()
  1. We will then bucketize the feature columns into two buckets around the mean since the TFBT estimator only takes bucketed features with the following code:
bc_fn = tf.feature_column.bucketized_columnnc_fn = tf.feature_column.numeric_columnbucketized_features = [bc_fn(source_column=nc_fn(key=column),                             boundaries=[x_train[column].mean()])                       for column ...

Get TensorFlow Machine Learning Projects now with O’Reilly online learning.

O’Reilly members experience live online training, plus books, videos, and digital content from 200+ publishers.