是否可以在 Golang 中腌制结构的实例

Nic*_*ler 5 pickle go

我正在用 Golang 做一些机器学习。我现在碰壁了,我训练的分类器需要将近半分钟的时间来训练,我想保存分类器的那个实例,这样我就不必每次都从头开始训练。Golang 应该怎么做呢? 仅供参考,我的分类器是一个结构体


当我用 python 做这类事情时,用 pickle 很容易。有等价物吗?

小智 6

尝试gobencoding/json来编组您的对象。之后,您可以将字符串存储到文件中。

下面是使用 json 的示例:

package main

import (
    "encoding/json"
     "fmt"
     "os"
)

type Book struct {
    Title string
    Pages []*Page
}

type Page struct {
    pageNumber int // remember to Capitalize the fields you want to marshal
    Content    string
}

func main() {
    // First make a book with nested array of struct pointers
    myBook := &Book{Title: "this is a title", Pages: make([]*Page, 0)}
    for i := 0; i < 3; i++ {
        myBook.Pages = append(myBook.Pages, &Page{i + 1, "words"})
    }

    // Open a file and dump JSON to it!
    f1, err := os.Create("/tmp/file1")
    enc := json.NewEncoder(f1)
    err = enc.Encode(myBook)
    if err != nil {
        panic(err)
    }
    f1.Close()

    // Open the file and load the object back!
    f2, err := os.Open("/tmp/file1")
    dec := json.NewDecoder(f2)
    var v Book
    err = dec.Decode(&v)
    if err != nil {
        panic(err)
    }
    f2.Close()

    // Check
    fmt.Println(v.Title)            // Output: <this is a title>
    fmt.Println(v.Pages[1].Content) // Output: <words>

    // pageNumber is not capitalized so it was not marshaled
    fmt.Println(v.Pages[1].pageNumber) // Output: <0>

}
Run Code Online (Sandbox Code Playgroud)