OrgLion-ML/golearn
1
0
Fork 0
mirror of https://github.com/sjwhitworth/golearn.git synced 2025-04-26 13:49:14 +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 {
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 {
@ -144,7 +148,6 @@ func (d *DecisionTreeNode) SaveWithPrefix(writer *base.ClassifierSerializer, pre
return err
}
writer.Close()
return nil
}