go-ml.dev/pkg/base@v0.0.0-20200610162856-60c38abac71b/model/dataset.go (about)

     1  package model
     2  
     3  import (
     4  	"go-ml.dev/pkg/base/tables"
     5  )
     6  
     7  /*
     8  Dataset is an abstraction of some source of a data to feed hungry models
     9  */
    10  type Dataset struct {
    11  	Source     tables.AnyData // It can be tables.Table or lazy stream of mlutil.Struct objects
    12  	Validation tables.AnyData // optional, equal to Source if nil
    13  	Label      string         // name of float32/Tensor field containing label to train
    14  	Test       string         // name of boolean field to select test data
    15  	Features   []string       // patterns of feature names to train model or predict
    16  }