Files
tensorneat-mend/problems/function_fitting/xor.py
2023-05-10 19:30:12 +08:00

14 lines
450 B
Python

import numpy as np
from . import FunctionFittingProblem
class Xor(FunctionFittingProblem):
def __init__(self):
self.num_inputs = 2
self.num_outputs = 1
self.batch = 4
self.inputs = np.array([[0, 0], [0, 1], [1, 0], [1, 1]], dtype=np.float32)
self.target = np.array([[0], [1], [1], [0]], dtype=np.float32)
super().__init__(self.num_inputs, self.num_outputs, self.batch, self.inputs, self.target)