sample

package

v1.0.2 Latest Latest Go to latest Published: May 8, 2025 License: MIT Imports: 4 Imported by: 0

Details

This section is empty.

This section is empty.

This section is empty.

type Reader struct {
	// contains filtered or unexported fields
}

Reader sample reader

func NewReader(r io.ReadSeeker) (*Reader, error)

NewReader create sample reader

func (r *Reader) BatchSize() uint32

BatchSize get batch size

func (r *Reader) FeatureSize() uint32

FeatureSize get feature size

func (r *Reader) LabelSize() uint32

LabelSize get label size

func (r *Reader) ReadSample(idx uint32, features, labels []float32) error

ReadSample read sample data

type Writer struct {
	// contains filtered or unexported fields
}

Writer sample writer

func NewWriter(w io.WriteSeeker) *Writer

NewWriter create sample writer

func (w *Writer) Close() error

Close rewrite header

func (w *Writer) WriteSample(features, labels []float32) error

WriteSample write sample data