Update config.pbtxt
Browse files- config.pbtxt +1 -1
config.pbtxt
CHANGED
|
@@ -24,7 +24,7 @@ output [
|
|
| 24 |
{
|
| 25 |
name: "predicted_class"
|
| 26 |
data_type: TYPE_INT64
|
| 27 |
-
dims: [ ] # Binary classification
|
| 28 |
}
|
| 29 |
]
|
| 30 |
|
|
|
|
| 24 |
{
|
| 25 |
name: "predicted_class"
|
| 26 |
data_type: TYPE_INT64
|
| 27 |
+
dims: [ 1 ] # Binary classification
|
| 28 |
}
|
| 29 |
]
|
| 30 |
|