v1

type DatasetInterface interface {
	Create(name, trainData, trainLabels, testData, testLabels string) error
	Delete(name string) error
	Get(name string) (*api.DatasetSummary, error)
	List() ([]api.DatasetSummary, error)
}

type DatasetsGetter interface {
	Datasets() DatasetInterface
}

type HistoryGetter interface {
	Histories() HistoryInterface
}

type HistoryInterface interface {
	Get(taskId string) (*api.History, error)
	Delete(taskId string) error
	List() ([]api.History, error)
	Prune() error
}

type NetworkGetter interface {
	Networks() NetworkInterface
}

type NetworkInterface interface {
	Train(req *api.TrainRequest) (string, error)
	Infer(req *api.InferRequest) ([]byte, error)
}

type TaskGetter interface {
	Tasks() TaskInterface
}

type TaskInterface interface {
	List() ([]api.TrainTask, error)
	Stop(id string) error
}

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

type V1Interface interface {
	NetworkGetter
	DatasetsGetter
	HistoryGetter
	TaskGetter
}