예제 #1
0
// 使用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)
}
예제 #2
0
// 从指定的目录读取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
}