Skip to content

Commit 08a6ae6

Browse files
committed
saved stash backup
1 parent a22ab5a commit 08a6ae6

File tree

1 file changed

+41
-0
lines changed

1 file changed

+41
-0
lines changed

stash.diff

+41
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,41 @@
1+
diff --git a/tsaplay/utils/addons.py b/tsaplay/utils/addons.py
2+
index fef0c92..4203b11 100644
3+
--- a/tsaplay/utils/addons.py
4+
+++ b/tsaplay/utils/addons.py
5+
@@ -146,7 +146,7 @@ def conf_matrix(model, features, labels, spec, params):
6+
depth=params["_n_out_classes"],
7+
),
8+
num_classes=params["_n_out_classes"],
9+
- )
10+
+ ),
11+
}
12+
)
13+
eval_hooks += [
14+
@@ -217,7 +217,26 @@ def scalars(model, features, labels, spec, params):
15+
),
16+
"auc": tf.metrics.auc(
17+
labels=tf.one_hot(indices=labels, depth=params["_n_out_classes"]),
18+
- predictions=spec.predictions["probabilities"],
19+
+ predictions=tf.Print(
20+
+ input_=spec.predictions["probabilities"],
21+
+ data=[
22+
+ spec.predictions["probabilities"],
23+
+ tf.concat(
24+
+ [
25+
+ features["left"],
26+
+ features["target"],
27+
+ features["right"],
28+
+ ],
29+
+ axis=1,
30+
+ ),
31+
+ # features["left"],
32+
+ tf.expand_dims(features["target"], axis=2),
33+
+ # features["right"],
34+
+ labels,
35+
+ ],
36+
+ message="",
37+
+ summarize=5000,
38+
+ ),
39+
name="auc_op",
40+
),
41+
}

0 commit comments

Comments
 (0)