Python VisualPlaceCell Beispiele

Beispiel #1

Datei: test.py Projekt: NB-DQN/place_cell

from path_integrating_cell import PathIntegratingCell
from visual_place_cell import VisualPlaceCell

import numpy as np

# c = PathIntegratingCell((9, 9))
#
# c.move(0, (0, 0))
# c.move(1, (1, 0))
# c.move(2, (0, 0))

c = VisualPlaceCell((9, 9))
c.move(0, np.array([0, 0, 0, 0, 0, 0, 0, 1, 1, 0, 0, 0, 0, 0, 0, 0, 1, 1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 1, 1, 1, 1, 1, 1, 1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 1]))
print(c.virtual_coordinate)
c.move(1, 0)
print(c.virtual_coordinate)
c.move(2, 0)
print(c.virtual_coordinate)

Beispiel #2

Datei: test_visual_output.py Projekt: NB-DQN/place_cell

from visual_place_cell import VisualPlaceCell
from deterministic_place_cell import DeterministicPlaceCell

import random
from matplotlib import pyplot as plt

size = (9, 9)

res = []
for j in range(1000):
    vpc = VisualPlaceCell(size)
    dpc = DeterministicPlaceCell(size)
    for i in range(100):
        while True:
            a = random.choice(range(4))
            if dpc.validate_action(a):
                break

        vpc.move(a)
        dpc.move(a)

        if vpc.virtual_coordinate != dpc.virtual_coordinate:
            res.append(i)
            break
    else:
        res.append(100)
        pass

    print(j)

plt.hist(res, bins=100)