def __init__(self):
QtGui.QWidget.__init__(self)
self.net = SkyNet(
nb_input=3, nb_hidden=2, nb_output=1,
eta=0.5, alpha=0.1,
)
teacher = ColorTeacher(self.net)
teacher.train(100000, verbose=True)
layout = QtGui.QVBoxLayout(self)
buttons_layout = QtGui.QHBoxLayout()
layout.addLayout(buttons_layout)
color = rand_color()
self.inputs = normalize(color)
self.black_button = QtGui.QPushButton('BLACK', self)
self.black_button.clicked.connect(self.on_black)
self.update_button_style('black', color)
buttons_layout.addWidget(self.black_button)
self.white_button = QtGui.QPushButton('WHITE', self)
self.white_button.clicked.connect(self.on_white)
self.update_button_style('white', color)
buttons_layout.addWidget(self.white_button)
self.auto_button = QtGui.QPushButton("BRAIN (error %.5f)" % 0.0, self)
self.auto_button.clicked.connect(self.on_auto)
layout.addWidget(self.auto_button)
self.label = QtGui.QLabel("Result: 0, Choice: white")
layout.addWidget(self.label)
def test_xor():
patterns = [
[[0,0], [0]],
[[0,1], [1]],
[[1,0], [1]],
[[1,1], [0]]
]
net = SkyNet(
nb_input=2, nb_hidden=4, nb_output=1,
eta=0.3, alpha=0.01,
)
teacher = PatternTeacher(net, patterns)
teacher.train(n=10000, verbose=True)
print("\nTest the trained network:\n")
for inputs, targets in patterns:
print(" Input:\t%s" % inputs)
print(" Expected:\t%s" % targets)
print(" Output:\t%s" % net.feed_forward(inputs))
print("")
def __init__(self):
QtGui.QWidget.__init__(self)
self.net = SkyNet(
nb_input=3,
nb_hidden=2,
nb_output=1,
eta=0.5,
alpha=0.1,
)
teacher = ColorTeacher(self.net)
teacher.train(100000, verbose=True)
layout = QtGui.QVBoxLayout(self)
buttons_layout = QtGui.QHBoxLayout()
layout.addLayout(buttons_layout)
color = rand_color()
self.inputs = normalize(color)
self.black_button = QtGui.QPushButton('BLACK', self)
self.black_button.clicked.connect(self.on_black)
self.update_button_style('black', color)
buttons_layout.addWidget(self.black_button)
self.white_button = QtGui.QPushButton('WHITE', self)
self.white_button.clicked.connect(self.on_white)
self.update_button_style('white', color)
buttons_layout.addWidget(self.white_button)
self.auto_button = QtGui.QPushButton("BRAIN (error %.5f)" % 0.0, self)
self.auto_button.clicked.connect(self.on_auto)
layout.addWidget(self.auto_button)
self.label = QtGui.QLabel("Result: 0, Choice: white")
layout.addWidget(self.label)
class Window(QtGui.QWidget):
def __init__(self):
QtGui.QWidget.__init__(self)
self.net = SkyNet(
nb_input=3, nb_hidden=2, nb_output=1,
eta=0.5, alpha=0.1,
)
teacher = ColorTeacher(self.net)
teacher.train(100000, verbose=True)
layout = QtGui.QVBoxLayout(self)
buttons_layout = QtGui.QHBoxLayout()
layout.addLayout(buttons_layout)
color = rand_color()
self.inputs = normalize(color)
self.black_button = QtGui.QPushButton('BLACK', self)
self.black_button.clicked.connect(self.on_black)
self.update_button_style('black', color)
buttons_layout.addWidget(self.black_button)
self.white_button = QtGui.QPushButton('WHITE', self)
self.white_button.clicked.connect(self.on_white)
self.update_button_style('white', color)
buttons_layout.addWidget(self.white_button)
self.auto_button = QtGui.QPushButton("BRAIN (error %.5f)" % 0.0, self)
self.auto_button.clicked.connect(self.on_auto)
layout.addWidget(self.auto_button)
self.label = QtGui.QLabel("Result: 0, Choice: white")
layout.addWidget(self.label)
def update_button_style(self, text, background):
button = getattr(self, text + "_button")
button.setStyleSheet("""
background-color: rgb%s;
color: %s;
border: none;
padding: 50px;
""" % (background, text))
def train(self, inputs, targets):
self.net.feed_forward(inputs)
error = self.net.back_propagate(targets)
self.auto_button.setText("BRAIN (error %.5f)" % error)
def on_black(self):
color = rand_color()
self.inputs = normalize(color)
self.update_button_style('black', color)
self.update_button_style('white', color)
self.train(self.inputs, [1.0])
def on_white(self):
color = rand_color()
self.inputs = normalize(color)
self.update_button_style('black', color)
self.update_button_style('white', color)
self.train(self.inputs, [0.0])
def on_auto(self):
result = self.net.feed_forward(self.inputs)[0]
self.label.setText("Result: %.2f, Choice: %s, Brighness: %.2f" % (
result, ("white" if result < 0.5 else "black"),
brightness(self.inputs),
))
class Window(QtGui.QWidget):
def __init__(self):
QtGui.QWidget.__init__(self)
self.net = SkyNet(
nb_input=3,
nb_hidden=2,
nb_output=1,
eta=0.5,
alpha=0.1,
)
teacher = ColorTeacher(self.net)
teacher.train(100000, verbose=True)
layout = QtGui.QVBoxLayout(self)
buttons_layout = QtGui.QHBoxLayout()
layout.addLayout(buttons_layout)
color = rand_color()
self.inputs = normalize(color)
self.black_button = QtGui.QPushButton('BLACK', self)
self.black_button.clicked.connect(self.on_black)
self.update_button_style('black', color)
buttons_layout.addWidget(self.black_button)
self.white_button = QtGui.QPushButton('WHITE', self)
self.white_button.clicked.connect(self.on_white)
self.update_button_style('white', color)
buttons_layout.addWidget(self.white_button)
self.auto_button = QtGui.QPushButton("BRAIN (error %.5f)" % 0.0, self)
self.auto_button.clicked.connect(self.on_auto)
layout.addWidget(self.auto_button)
self.label = QtGui.QLabel("Result: 0, Choice: white")
layout.addWidget(self.label)
def update_button_style(self, text, background):
button = getattr(self, text + "_button")
button.setStyleSheet("""
background-color: rgb%s;
color: %s;
border: none;
padding: 50px;
""" % (background, text))
def train(self, inputs, targets):
self.net.feed_forward(inputs)
error = self.net.back_propagate(targets)
self.auto_button.setText("BRAIN (error %.5f)" % error)
def on_black(self):
color = rand_color()
self.inputs = normalize(color)
self.update_button_style('black', color)
self.update_button_style('white', color)
self.train(self.inputs, [1.0])
def on_white(self):
color = rand_color()
self.inputs = normalize(color)
self.update_button_style('black', color)
self.update_button_style('white', color)
self.train(self.inputs, [0.0])
def on_auto(self):
result = self.net.feed_forward(self.inputs)[0]
self.label.setText("Result: %.2f, Choice: %s, Brighness: %.2f" % (
result,
("white" if result < 0.5 else "black"),
brightness(self.inputs),
))