📝 Add commens about HDF5 datatype

vdaas · Jan 25, 2021 · 068fb26 · 068fb26
1 parent 0fdcb0c
commit 068fb26
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/example/client/agent/main.go b/example/client/agent/main.go
@@ -171,6 +171,8 @@ func load(path string) (ids []string, train, test [][]float32, err error) {
 		row, dim := int(dims[0]), int(dims[1])
 
 		// Gets the stored vector. All are represented as one-dimensional arrays.
+		// The type of the slice depends on your dataset.
+		// For fashion-mnist-784-euclidean.hdf5, the datatype is float32.
 		vec := make([]float32, sp.SimpleExtentNPoints())
 		if err := d.Read(&vec); err != nil {
 			return nil, err

diff --git a/example/client/main.go b/example/client/main.go
@@ -153,6 +153,8 @@ func load(path string) (ids []string, train, test [][]float32, err error) {
 		row, dim := int(dims[0]), int(dims[1])
 
 		// Gets the stored vector. All are represented as one-dimensional arrays.
+		// The type of the slice depends on your dataset.
+		// For fashion-mnist-784-euclidean.hdf5, the datatype is float32.
 		vec := make([]float32, sp.SimpleExtentNPoints())
 		if err := d.Read(&vec); err != nil {
 			return nil, err