// 使用JSON格式加载
func (t *FileStorage) Load(target interface{}) error {
// 不存在存储的数据, 跳过加载
ok := util.IsExist(t.path)
if ok == false {
return nil
}
// 加载JSON数据
stData, err := util.ReadFile(t.path)
if err != nil {
return err
}
return json.Unmarshal(stData, target)
}
// 从指定的目录读取txt文件进行训练
func FileTrain(path string, classifier *Classifier) (int, error) {
fs, err := util.ReadDir(path)
if err != nil {
return 0, err
}
i := 0
for _, f := range fs {
doc, err := util.ReadFile(path + "/" + f.Name())
if err != nil {
continue
}
category := getCategory(doc)
if len(category) == 0 {
continue
}
content := doc[len(category):]
classifier.Training(string(content), string(category))
i++
}
return i, nil
}