Documentation ¶
Overview ¶
Copyright 2018 kurosawa. All Rights Reserved.
Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance with the License. You may obtain a copy of the License at
http://www.apache.org/licenses/LICENSE-2.0
Unless required by applicable law or agreed to in writing, software distributed under the License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions and limitations under the License. =============================================================================
Copyright 2018 kurosawa. All Rights Reserved.
Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance with the License. You may obtain a copy of the License at
http://www.apache.org/licenses/LICENSE-2.0
Unless required by applicable law or agreed to in writing, software distributed under the License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions and limitations under the License. =============================================================================
Copyright 2018 kurosawa. All Rights Reserved.
Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance with the License. You may obtain a copy of the License at
http://www.apache.org/licenses/LICENSE-2.0
Unless required by applicable law or agreed to in writing, software distributed under the License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions and limitations under the License. =============================================================================
Index ¶
- func Accuracy(x gmat.Tensor, t gmat.Tensor) float64
- func Allreduce(layer []LayerInterface)
- func AvePrint(x gmat.Tensor, name string)
- func BackLayer(layer []LayerInterface, dout gmat.Tensor) gmat.Tensor
- func DensePrint(x gmat.Tensor, name string)
- func ForwardLayer(layer []LayerInterface, x gmat.Tensor, t gmat.Tensor) gmat.Tensor
- func ImageRead2GrayNorm(imgpath string) (t gmat.Tensor)
- func LayerAdd(layer *[]LayerInterface, calc interface{}, shape ...interface{})
- func Layerinit() interface{}
- func MomentumUpdateLayer(layer []LayerInterface, m *Momentum)
- func OneHot(x int, size int) (y []float64)
- func Predict(layers []LayerInterface, x gmat.Tensor) int
- func Restore(fileName string, p *[]LayerInterface)
- func Run(layer []LayerInterface, update interface{}, x gmat.Tensor, t gmat.Tensor) (loss gmat.Tensor)
- func SGDUpdateLayer(layer []LayerInterface, sgd *SGD)
- func Saver(p []LayerInterface, saveName string)
- type Dense
- type Dropout
- type LayerInterface
- type Momentum
- type Relu
- type SGD
- type Sigmoid
- type SoftmaxWithLoss
Constants ¶
This section is empty.
Variables ¶
This section is empty.
Functions ¶
func Allreduce ¶
func Allreduce(layer []LayerInterface)
func DensePrint ¶
func ForwardLayer ¶
func ImageRead2GrayNorm ¶
func LayerAdd ¶
func LayerAdd(layer *[]LayerInterface, calc interface{}, shape ...interface{})
func MomentumUpdateLayer ¶
func MomentumUpdateLayer(layer []LayerInterface, m *Momentum)
func Restore ¶
func Restore(fileName string, p *[]LayerInterface)
func SGDUpdateLayer ¶
func SGDUpdateLayer(layer []LayerInterface, sgd *SGD)
func Saver ¶
func Saver(p []LayerInterface, saveName string)
Types ¶
type LayerInterface ¶
type SoftmaxWithLoss ¶
type SoftmaxWithLoss layer.SoftmaxWithLoss