You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
29 lines
1021 B
29 lines
1021 B
#!/usr/bin/env python
|
|
|
|
from lib.func import AIlib as ai
|
|
|
|
class rgb(object):
|
|
def __init__(self, loadedWeights = None, loadedBias = None):
|
|
|
|
if( not loadedWeights or not loadedBias ):
|
|
self.weights = [ ai.genRandomMatrix(3, 4), ai.genRandomMatrix(4, 4), ai.genRandomMatrix(4, 3) ] # array of matrices of weights
|
|
# 3 input neurons -> 4 hidden neurons -> 4 hidden neurons -> 3 output neurons
|
|
|
|
# Will be needing biases too
|
|
self.bias = [ ai.genRandomMatrix(1, 4), ai.genRandomMatrix(1, 4), ai.genRandomMatrix(1, 3) ]
|
|
# This doesn't look very good, but it works so...
|
|
# This is all we need
|
|
else: # if we want to load our progress from before then this would do it
|
|
print("Loading neural net...")
|
|
self.weights = loadedWeights
|
|
self.bias = loadedBias
|
|
|
|
def think(self, inputMatrix):
|
|
print(self.weights)
|
|
print(self.bias)
|
|
def init(): # init func
|
|
bot = rgb()
|
|
|
|
bot.think(1)
|
|
|
|
init()
|
|
|