Update fraud example

ospillinger · ospillinger · commit a327f34caf54 · 2019-04-03T18:29:55.000-07:00
diff --git a/examples/fraud/implementations/models/dnn.py b/examples/fraud/implementations/models/dnn.py
@@ -11,6 +11,6 @@ def create_estimator(run_config, model_config):
         feature_columns=feature_columns,
         hidden_units=model_config["hparams"]["hidden_units"],
         n_classes=2,
-        weight_column="class_weight",
+        weight_column="weight_column",
         config=run_config,
     )
diff --git a/examples/fraud/resources/dnn.yaml b/examples/fraud/resources/dnn.yaml
@@ -9,7 +9,7 @@
      v15_normalized, v16_normalized, v17_normalized, v18_normalized, v19_normalized,
      v20_normalized, v21_normalized, v22_normalized, v23_normalized, v24_normalized,
      v25_normalized, v26_normalized, v27_normalized, v28_normalized, amount_normalized]
-  training_columns: [class_weight]
+  training_columns: [weight_column]
   hparams:
     hidden_units: [100, 100, 100]
   data_partition_ratio:
diff --git a/examples/fraud/resources/raw_columns.yaml b/examples/fraud/resources/raw_columns.yaml
diff --git a/examples/fraud/resources/weight_column.yaml b/examples/fraud/resources/weight_column.yaml
@@ -15,7 +15,7 @@
   output_type: FLOAT_COLUMN
 
 - kind: transformed_column
-  name: class_weight
+  name: weight_column
   transformer: weight
   inputs:
     columns:

Original file line number	Diff line number	Diff line change
`@@ -11,6 +11,6 @@ def create_estimator(run_config, model_config):`
`11`	`11`	`feature_columns=feature_columns,`
`12`	`12`	`hidden_units=model_config["hparams"]["hidden_units"],`
`13`	`13`	`n_classes=2,`
`14`		`- weight_column="class_weight",`
	`14`	`+ weight_column="weight_column",`
`15`	`15`	`config=run_config,`
`16`	`16`	`)`