OrgLion-ML/golearn
1
0
Fork 0
mirror of https://github.com/sjwhitworth/golearn.git synced 2025-04-28 13:48:56 +08:00
Code Issues Releases Wiki Activity

trees: Try to fix premature write-after-Close issue

Browse Source
This commit is contained in:
Richard Townsend 2018-01-28 16:35:55 +00:00
parent e2279995c1
commit 58ae6f4d1b
1 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
trees/id3.go
View File

@ -126,7 +126,11 @@ func (d *DecisionTreeNode) Save(filePath string) error {
if err != nil { if err != nil {
return err return err
} }
return d.SaveWithPrefix(serializer, "") err = d.SaveWithPrefix(serializer, "")
if err != nil {
return err
}
return serializer.Close()
} }
func (d *DecisionTreeNode) SaveWithPrefix(writer *base.ClassifierSerializer, prefix string) error { func (d *DecisionTreeNode) SaveWithPrefix(writer *base.ClassifierSerializer, prefix string) error {
@ -144,7 +148,6 @@ func (d *DecisionTreeNode) SaveWithPrefix(writer *base.ClassifierSerializer, pre
return err return err
} }
writer.Close()
return nil return nil
} }