Skip to content

Commit fef5fae

Browse files
committed
Remove my_model.py from init CLI command
1 parent 92737c6 commit fef5fae

File tree

1 file changed

+0
-36
lines changed

1 file changed

+0
-36
lines changed

cli/cmd/init.go

Lines changed: 0 additions & 36 deletions
Original file line numberDiff line numberDiff line change
@@ -186,42 +186,6 @@ func appInitFiles(appName string) map[string]string {
186186
}
187187
`,
188188

189-
"implementations/models/my_model.py": `import tensorflow as tf
190-
191-
192-
def create_estimator(run_config, model_config):
193-
"""Create an estimator to train the model.
194-
195-
Args:
196-
run_config: An instance of tf.estimator.RunConfig to be used when creating
197-
the estimator.
198-
199-
model_config: The Cortex configuration for the model. Column references in all
200-
inputs (i.e. model_config["target_column"], model_config["input"], and
201-
model_config["training_input"]) are replaced by their names (e.g. "@column1"
202-
will be replaced with "column1"). All other resource references (e.g. constants
203-
and aggregates) are replaced by their runtime values.
204-
205-
Returns:
206-
An instance of tf.estimator.Estimator to train the model.
207-
"""
208-
209-
## Sample create_estimator implementation:
210-
#
211-
# feature_columns = []
212-
# for col_name in model_config["input"]["numeric_columns"]:
213-
# feature_columns.append(tf.feature_column.numeric_column(col_name))
214-
#
215-
# return tf.estimator.DNNClassifier(
216-
# feature_columns=feature_columns,
217-
# n_classes=model_config["input"]["num_classes"],
218-
# hidden_units=model_config["hparams"]["hidden_units"],
219-
# config=run_config,
220-
# )
221-
222-
pass
223-
`,
224-
225189
"resources/constants.yaml": `## Sample constant:
226190
#
227191
# - kind: constant

0 commit comments

Comments
 (0)