Skip to content

Commit 8e4a98a

Browse files
committed
Merge branch 'tf.keras-0.3.image-classification'
2 parents 4eb9323 + 8b6dc47 commit 8e4a98a

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

src/TensorFlowNET.Core/Data/DatasetV2.cs

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -132,7 +132,9 @@ public override string ToString()
132132
break;
133133
}
134134

135-
yield return (results[0], results.Length == 1 ? null : results[1]);
135+
yield return results.Length == 2
136+
? (results[0], results[1])
137+
: (null, results[0]);
136138
}
137139
}
138140

0 commit comments

Comments
 (0)