agent

package

v0.2.23-0...-4aaa12c Latest Latest Go to latest Published: May 17, 2020 License: NCSA Imports: 38 Imported by: 9

Details

Valid go.mod file

The Go module system was introduced in Go 1.11 and is the official dependency management solution for Go.
Redistributable license

Redistributable licenses place minimal restrictions on how software can be used, modified, and redistributed.
Tagged version

Modules with tagged versions give importers more predictable builds.
Stable version

When a project reaches major version v1 it is considered stable.
Learn more about best practices

Repository

github.com/rai-project/dlframework

Links

Open Source Insights

Documentation ¶

Index ¶

Variables
func AddPredictor(framework dl.FrameworkManifest, pred predictor.Predictor) error
func GetPredictor(framework dl.FrameworkManifest) (predictor.Predictor, error)
func GetPredictors(framework dl.FrameworkManifest) ([]predictor.Predictor, error)
func PredictorFrameworks() []dl.FrameworkManifest
func Predictors() []string
func UploadDir() (string, error)
type Agent
- func New(predictors []predictor.Predictor, opts ...Option) (*Agent, error)
type Option
type Options
- func NewOptions(opts ...Option) (*Options, error)
type PredictorLifetime
type Registry

Constants ¶

This section is empty.

Variables ¶

View Source

var (
	DefaultChannelBuffer = 1000
)

Functions ¶

func AddPredictor ¶

func AddPredictor(framework dl.FrameworkManifest, pred predictor.Predictor) error

func GetPredictor ¶

func GetPredictor(framework dl.FrameworkManifest) (predictor.Predictor, error)

func GetPredictors ¶

func GetPredictors(framework dl.FrameworkManifest) ([]predictor.Predictor, error)

func PredictorFrameworks ¶

func PredictorFrameworks() []dl.FrameworkManifest

func Predictors ¶

func Predictors() []string

func UploadDir ¶

func UploadDir() (string, error)

Types ¶

type Agent ¶

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

func New ¶

func New(predictors []predictor.Predictor, opts ...Option) (*Agent, error)

func (*Agent) Close ¶

func (p *Agent) Close(ctx context.Context, req *dl.PredictorCloseRequest) (*dl.PredictorCloseResponse, error)

Close a predictor and clear it's memory.

func (*Agent) Dataset ¶

func (p *Agent) Dataset(ctx context.Context, req *dl.DatasetRequest) (*dl.FeaturesResponse, error)

Dataset method receives a single dataset and runs the predictor on all elements of the dataset. The result is a prediction feature list.

func (*Agent) FindFrameworkModel ¶

func (p *Agent) FindFrameworkModel(ctx context.Context, req *dl.PredictorOpenRequest) (*dl.FrameworkManifest, *dl.ModelManifest, error)

func (*Agent) Images ¶

func (p *Agent) Images(ctx context.Context, req *dl.ImagesRequest) (*dl.FeaturesResponse, error)

Image method receives a list base64 encoded images and runs the predictor on all the images. The result is a prediction feature list for each image.

func (*Agent) Open ¶

func (p *Agent) Open(ctx context.Context, req *dl.PredictorOpenRequest) (*dl.Predictor, error)

Opens a predictor and returns an id where the predictor is accessible. The id can be used to perform inference requests.

func (*Agent) PublishInPredictor ¶

func (b *Agent) PublishInPredictor(host, prefix string) error

func (*Agent) PublishInRegistery ¶

func (b *Agent) PublishInRegistery(prefix string) error

func (*Agent) RegisterManifests ¶

func (p *Agent) RegisterManifests() (*grpc.Server, error)

func (*Agent) RegisterPredictor ¶

func (p *Agent) RegisterPredictor() (*grpc.Server, error)

func (*Agent) Reset ¶

func (p *Agent) Reset(ctx context.Context, req *dl.ResetRequest) (*dl.ResetResponse, error)

Clear method clears the internal cache of the predictors

func (*Agent) URLs ¶

func (p *Agent) URLs(ctx context.Context, req *dl.URLsRequest) (*dl.FeaturesResponse, error)

URLs method receives a list of urls and runs the predictor on all the urls. The result is a list of predicted features for all the urls.

type Option ¶

type Option func(o *Options) *Options

func WithHost ¶

func WithHost(host string) Option

func WithPort ¶

func WithPort(port int) Option

func WithPortString ¶

func WithPortString(port string) Option

type Options ¶

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

func NewOptions ¶

func NewOptions(opts ...Option) (*Options, error)

type PredictorLifetime ¶

type PredictorLifetime struct {
	Predictor      *predictor.Predictor
	ReferenceCount uint64
}

type Registry ¶

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

func (*Registry) FrameworkAgents ¶

func (c *Registry) FrameworkAgents(context.Context, *dl.FrameworkRequest) (*dl.Agents, error)

func (*Registry) FrameworkManifests ¶

func (c *Registry) FrameworkManifests(context.Context, *dl.FrameworkRequest) (*dl.FrameworkManifestsResponse, error)

func (*Registry) ModelAgents ¶

func (c *Registry) ModelAgents(context.Context, *dl.ModelRequest) (*dl.Agents, error)

func (*Registry) ModelManifests ¶

func (c *Registry) ModelManifests(context.Context, *dl.ModelRequest) (*dl.ModelManifestsResponse, error)

func (*Registry) PublishInPredictor ¶

func (b *Registry) PublishInPredictor(host, prefix string) error

func (*Registry) PublishInRegistery ¶

func (c *Registry) PublishInRegistery() error

Source Files ¶

View all Source files

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