Searched refs:feature_cols (Results 1 – 3 of 3) sorted by relevance
51 def _get_estimator(output_dir, feature_cols): argument65 dnn_feature_columns=feature_cols,
54 def _get_tfbt(output_dir, feature_cols): argument70 feature_columns=feature_cols,
100 def _create_linear_model(self, feature_cols, partitioner): argument106 feature_columns=feature_cols,