Commit a457886e authored by Alison Carrera's avatar Alison Carrera
Browse files

Added new contextual bandit algorithm and updated example notebook.

parent ed53456a
This diff is collapsed.
......@@ -153,7 +153,7 @@ class ONN(nn.Module):
class ONN_THS(ONN):
def __init__(self, features_size, max_num_hidden_layers, qtd_neuron_per_hidden_layer, n_classes, b=0.99, n=0.01,
s=0.2, e=[0.5, 0.35, 0.2, 0.1], use_cuda=False):
s=0.2, e=[0.5, 0.35, 0.2, 0.1, 0.05], use_cuda=False):
super().__init__(features_size, max_num_hidden_layers, qtd_neuron_per_hidden_layer, n_classes, b=b, n=n, s=s,
use_cuda=use_cuda)
self.e = e
......
......@@ -6,7 +6,7 @@ with open(path.join(this_directory, 'README.md'), encoding='utf-8') as f:
long_description = f.read()
setup(name='onn',
version='0.1.1',
version='0.1.3',
description='Online Neural Network',
url='https://github.com/alison-carrera/onn',
author='Alison Carrera',
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment