PitchHut logo
Log in / Sign up
Note
by notedance
Note is a machine learning library.
Pitch

Note is a machine learning library. Note makes the building and training of neural networks easy and flexible. Note can be used not only for deep learning but also for reinforcement learning, it allows you to easily train agents built with Note, Keras, or PyTorch using reinforcement learning.

Description

Deep learning:

Note.nn.layer package contains many layer modules, you can use them to build neural networks. Note’s layer modules are implemented based on TensorFlow, which means they are compatible with TensorFlow’s API. The layer modules allow you to build neural networks in the style of PyTorch or Keras. You can use the layer modules to build neural networks trained with TensorFlow.

https://github.com/NoteDance/Note/tree/Note-7.0/Note/nn/layer

Documentation: https://github.com/NoteDance/Note-documentation/tree/layer-7.0

Using Note’s Layer module, you can determine the shape of the training parameters when you input data like Keras, or you can give the shape of the training parameters in advance like PyTorch.

Pytorch:

from Note import nn

class model(nn.Model):
    def __init__(self):
	super().__init__()
        self.layer1=nn.dense(128,784,activation='relu')
        self.layer2=nn.dense(10,128)
    
    def __call__(self,data):
        x=self.layer1(data)
        x=self.layer2(x)
        return x

Keras:

from Note import nn

class model(nn.Model):
    def __init__(self):
	super().__init__()
        self.layer1=nn.dense(128,activation='relu')
        self.layer2=nn.dense(10)
    
    def __call__(self,data):
        x=self.layer1(data)
        x=self.layer2(x)
        return x

Note.models.tf package contains neural networks implemented with Note’s layer module that can be trained with TensorFlow. You can also consider these models as examples using the Note. The documentation shows how to train, test, save, and restore models built with Note.

https://github.com/NoteDance/Note/tree/Note-7.0/Note/models/tf

Documentation: https://github.com/NoteDance/Note-documentation/tree/Model-7.0

Reinforcement learning:

You just need to have your agent class inherit from the RL or RL_pytorch class, and you can easily train your agent built with Note, Keras or PyTorch. You can learn how to build an agent from the examples here. The documentation shows how to train, save, and restore agent built with Note, Keras or PyTorch.

Documentation: https://github.com/NoteDance/Note-documentation/tree/RL-7.0

Function modules:

Documentation: https://github.com/NoteDance/Note-documentation/tree/function-7.0

Note.nn.Model.Model:

Model class manages the parameters and layers of neural network.

Model's function documentation: https://github.com/NoteDance/Note-documentation/tree/function-7.0

0 comments

No comments yet.

Sign in to be the first to comment.