tstraining

package
v0.0.0-...-165e7e6 Latest Latest
Warning

This package is not in the latest version of its module.

Go to latest
Published: Jun 11, 2022 License: MIT Imports: 7 Imported by: 0

Documentation

Index

Constants

This section is empty.

Variables

This section is empty.

Functions

This section is empty.

Types

type DataTrainer

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

func NewDataTrainer

func NewDataTrainer(
	workingPath string,
	dryRun bool,
	containsIP bool,
	dataForPrediction bool,
	readDataHeader bool,
) *DataTrainer

func (*DataTrainer) Extract

func (d *DataTrainer) Extract(trafficStats *[]domain.TrafficStatistic) []domain.TsTrainingData

Extract extracts training data from PacketFlow struct

func (*DataTrainer) LoadFromJSON

func (d *DataTrainer) LoadFromJSON(absPath string) ([]domain.TsTrainingData, error)

LoadFromJSON returns trained data from a specified file. If read fails, then returns an empty array with error

func (*DataTrainer) ReadFromCSV

func (d *DataTrainer) ReadFromCSV() ([]domain.TsTrainingData, error)

ReadFromCSV reads from given path, and returns Training data It takes absPath and info if contains header Returns array of training data. If something go bad, then returns an empty array with error Not suitable for reading model that contain IP fields, only for trained model ready for production !

func (*DataTrainer) SaveAsCSV

func (d *DataTrainer) SaveAsCSV(data []domain.TsTrainingData, absPath string) error

SaveAsCSV takes a data and absolute path to a file Returns err while something go bad. If csv contains IP addr, then should not be used for training model, only for user pourpose

func (*DataTrainer) SaveAsJSON

func (d *DataTrainer) SaveAsJSON(data *[]domain.TsTrainingData, absPath string) error

SaveAsJSON saves training data into a txt file

Jump to

Keyboard shortcuts

? : This menu
/ : Search site
f or F : Jump to
y or Y : Canonical URL