我在Python中适合one hot encoder。例如,我可以使用泡菜或joblib保存它。 我想知道是否可以将其加载到golang中以预处理数据。
import (
"io"
"os"
"github.com/hydrogen18/stalecucumber"
prep "github.com/pa-m/sklearn/preprocessing"
"gonum.org/v1/gonum/mat"
)
func main() {
var ohe prep.OneHotEncoder
oheFilePath := "one_hot_encoder.pkl"
var r io.Reader
r, _ = os.Open(oheFilePath)
stalecucumber.UnpackInto(&ohe).From(stalecucumber.Unpickle(r))
}
然后我想对我的字符串向量进行一次热编码:
input := []string{"ios", "US"}
所以我会做类似的事情
ohe.Transform(input)
但是Transform函数需要两个mat.Matrix作为输入,我无法定义字符串矩阵,而且我也只有一个输入(我不编码任何标签)。