如何在golang中加载pythonsklearnone热编码器?

2024-05-29 02:43:45 发布

您现在位置:Python中文网/ 问答频道 /正文

我在Python中安装了一个one hot encoder。例如,我可以使用pickle或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)

但是转换函数需要两个材料矩阵作为输入,我不能定义字符串的矩阵,我也只有一个输入(我不编码任何标签)。你知道吗


Tags: 字符串iogithubcom编码encoderosvar

热门问题