TensorFlow Machine Learning Projects

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()

2. 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 ...