Need help with goro?
Click the “chat” button below for chat support from the developer who created it, or find similar developers for support.

About the developer

aunum
269 Stars 11 Forks Apache License 2.0 11 Commits 1 Opened issues

Description

A High-level Machine Learning Library for Go

Services available

!
?

Need anything else?

Contributors list

logo
GoDoc Go Report Card

Overview

Goro is a high-level machine learning library for Go built on Gorgonia. It aims to have the same feel as Keras.

Usage

import (
    . "github.com/aunum/goro/pkg/v1/model"
    "github.com/aunum/goro/pkg/v1/layer"
)

// create the 'x' input e.g. mnist image x := NewInput("x", []int{1, 28, 28})

// create the 'y' or expect output e.g. labels y := NewInput("y", []int{10})

// create a new sequential model with the name 'mnist' model, _ := NewSequential("mnist")

// add layers to the model model.AddLayers( layer.Conv2D{Input: 1, Output: 32, Width: 3, Height: 3}, layer.MaxPooling2D{}, layer.Conv2D{Input: 32, Output: 64, Width: 3, Height: 3}, layer.MaxPooling2D{}, layer.Conv2D{Input: 64, Output: 128, Width: 3, Height: 3}, layer.MaxPooling2D{}, layer.Flatten{}, layer.FC{Input: 128 * 3 * 3, Output: 100}, layer.FC{Input: 100, Output: 10, Activation: layer.Softmax}, )

// pick an optimizer optimizer := g.NewRMSPropSolver()

// compile the model with options model.Compile(xi, yi, WithOptimizer(optimizer), WithLoss(m.CrossEntropy), WithBatchSize(100), )

// fit the model model.Fit(xTrain, yTrain)

// use the model to predict an 'x' prediction, _ := model.Predict(xTest)

// fit the model with a batch model.FitBatch(xTrainBatch, yTrainBatch)

// use the model to predict a batch of 'x' prediction, _ = model.PredictBatch(xTestBatch)

Examples

See the examples folder for example implementations.

There are many examples in the reinforcement learning library Gold.

Docs

Each package contains a README explaining the usage, also see GoDoc.

Contributing

Please open an MR for any issues or feature requests.

Feel free to ping @pbarker on Gopher slack.

Roadmap

  • [ ] RNN
  • [ ] LSTM
  • [ ] Summary
  • [ ] Visualization

We use cookies. If you continue to browse the site, you agree to the use of cookies. For more information on our use of cookies please see our Privacy Policy.