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

Fixing compilation errors that break the library

Browse Source 
The file in question has numerous references to the package "base"
whilst also being in the same package. My guess is that this was a
utility function and wasn't vetted for correctness after being moved
here.
This commit is contained in:
Paul Freakn Baker 2020-11-27 06:07:20 -07:00
parent 7240e2c238
commit 7580cac059
1 changed files with 8 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
base/dataframe_go.go
View File

@ -5,31 +5,31 @@ import (
"reflect"
"strconv"
dataframe "github.com/rocketlaunchr/dataframe-go"
"github.com/rocketlaunchr/dataframe-go"
)
// ConvertDataFrameToInstances converts a DataFrame-go dataframe object to Golearn Fixed Data Grid. Allows for compabitibility between dataframe and golearn's ML models.
// df is the dataframe Object. classAttrIndex is the index of the class Attribute in the data.i
func ConvertDataFrameToInstances(df *dataframe.DataFrame, classAttrIndex int) base.FixedDataGrid {
func ConvertDataFrameToInstances(df *dataframe.DataFrame, classAttrIndex int) FixedDataGrid {
// Creating Attributes based on Datafraem
// Creating Attributes based on Dataframe
names := df.Names()
attrs := make([]base.Attribute, len(names))
attrs := make([]Attribute, len(names))
newInst := base.NewDenseInstances()
newInst := NewDenseInstances()
for i := range names {
col := df.Series[i]
if reflect.TypeOf(col.Value(0)).Kind() == reflect.String {
attrs[i] = new(base.CategoricalAttribute)
attrs[i] = new(CategoricalAttribute)
attrs[i].SetName(names[i])
} else {
attrs[i] = base.NewFloatAttribute(names[i])
attrs[i] = NewFloatAttribute(names[i])
}
}
// Add the attributes
newSpecs := make([]base.AttributeSpec, len(attrs))
newSpecs := make([]AttributeSpec, len(attrs))
for i, a := range attrs {
newSpecs[i] = newInst.AddAttribute(a)
@ -60,7 +60,6 @@ func ConvertDataFrameToInstances(df *dataframe.DataFrame, classAttrIndex int) ba
newInst.Set(newSpecs[j], i, newSpecs[j].GetAttribute().GetSysValFromString(val))
}
}
return newInst