1package main
 2
 3import (
 4	"io/ioutil"
 5)
 6
 7var dataset map[string][]string
 8var categDataset []string
 9
10func readDataset(path string) {
11	dataset = make(map[string][]string)
12
13	folders, _ := ioutil.ReadDir(path)
14	for _, folder := range folders {
15		categDataset = append(categDataset, folder.Name())
16		folderFiles, _ := ioutil.ReadDir(path + "/" + folder.Name())
17		for _, file := range folderFiles {
18			dataset[folder.Name()] = append(dataset[folder.Name()], file.Name())
19		}
20	}
21}