如何在golang中加载Python sklearn一个热编码器?

时间:2019-05-17 10:22:41

标签: python go scikit-learn porting one-hot-encoding

我在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作为输入,我无法定义字符串矩阵,而且我也只有一个输入(我不编码任何标签)。

0 个答案:

没有答案