Как загрузить Python sklearn одним горячим кодером в golang? - PullRequest
Новая
       33

Как загрузить Python sklearn одним горячим кодером в golang?

0 голосов
/ 17 мая 2019

Я установил один горячий кодер в Python.Я могу сохранить его, например, используя 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)

Но функция Transform принимает два mat.Matrix в качестве входных данных, и я не могу определить матрицу строк, и у меня также есть только один вход (я не кодирую какие-либо метки).

...