import "github.com/quan-to/go-vsm/vsm"

vs := vsm.New(nil)

docs := []vsm.Document{
        {
                Sentence: "Shipment of gold damaged in a fire.",
                Class:    "d1",
        },
        {
                Sentence: "Delivery of silver arrived in a silver truck.",
                Class:    "d2",
        },
        {
                Sentence: "Shipment of gold arrived in a truck.",
                Class:    "d3",
        },
        ...,
}


// Statically training
for _, doc := range docs {
        if err := vs.StaticTraining(doc); err != nil {
                // Error occurred during training.
        }
}

docs := []vsm.Document{
        {
                Sentence: "Shipment of gold damaged in a fire.",
                Class:    "d1",
        },
        {
                Sentence: "Shipment of gold arrived in a truck.",
                Class:    "d3",
        },
}

vs := vsm.New(nil)

for _, doc := range docs {
        err := vs.StaticTraining(doc)
        fmt.Println(err)
}

docCh := make(chan Document)

go func() {
        defer close(docCh)

        // Loads document from some source dynamically
        // and sends it to the training channel.
        docCh <- vsm.Document{
                Sentence: "Delivery of silver arrived in a silver truck.",
                Class:    "d2",
        }
}()

trainCh := vs.DynamicTraining(context.Background(), docCh)

// Checks if error occurred during the training process.
for {
        res, ok := <-trainCh
        // trainCh closed. All train data was consumed.
        if !ok {
                break
        }

        if res.Err != nil {
                // Handles error.
        }
}

doc, err := vs.Search("gold silver truck.")

fmt.Println(doc.Class, err)