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

Merge pull request #142 from defcube/arff3

Browse Source 
Serialize ARFF to Writer in Addition to a File
This commit is contained in:
Richard Townsend 2016-07-04 13:23:46 +01:00 committed by GitHub
commit 4a2b2a30fc
1 changed files with 11 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
base/arff.go
View File

@ -37,8 +37,14 @@ func SerializeInstancesToDenseARFFWithAttributes(inst FixedDataGrid, rawAttrs []
}
defer f.Close()
return SerializeInstancesToWriterDenseARFFWithAttributes(f, inst, rawAttrs, relation)
}
func SerializeInstancesToWriterDenseARFFWithAttributes(w io.Writer, inst FixedDataGrid, rawAttrs []Attribute, relation string) error {
// Write @relation header
f.WriteString(fmt.Sprintf("@relation %s\n\n", relation))
fmt.Fprintf(w, "@relation %s\n\n", relation)
// Get all Attribute specifications
attrs := ResolveAttributes(inst, rawAttrs)
@ -51,17 +57,17 @@ func SerializeInstancesToDenseARFFWithAttributes(inst FixedDataGrid, rawAttrs []
vals := a.GetValues()
t = fmt.Sprintf("{%s}", strings.Join(vals, ", "))
}
f.WriteString(fmt.Sprintf("@attribute %s %s\n", attr.GetName(), t))
fmt.Fprintf(w, "@attribute %s %s\n", attr.GetName(), t)
}
f.WriteString("\n@data\n")
fmt.Fprint(w, "\n@data\n")
buf := make([]string, len(attrs))
inst.MapOverRows(attrs, func(val [][]byte, row int) (bool, error) {
for i, v := range val {
buf[i] = attrs[i].attr.GetStringFromSysVal(v)
}
f.WriteString(strings.Join(buf, ","))
f.WriteString("\n")
fmt.Fprint(w, strings.Join(buf, ","))
fmt.Fprint(w, "\n")
return true, nil
})