diff options
Diffstat (limited to 'src')
| -rw-r--r-- | src/graph.py | 6 | ||||
| -rw-r--r-- | src/scalar.py (renamed from src/value.py) | 8 |
2 files changed, 7 insertions, 7 deletions
diff --git a/src/graph.py b/src/graph.py index 2067071..9aecde1 100644 --- a/src/graph.py +++ b/src/graph.py @@ -4,10 +4,10 @@ from graphviz import Digraph import matplotlib.pyplot as plt import matplotlib.image as mpimg -from .value import Value +from .scalar import Scalar class Graph: - def __init__(self, root: Value) -> None: + def __init__(self, root: Scalar) -> None: self.dot = Digraph(format='png', graph_attr={ 'rankdir': 'LR' }) self.nodes = set() @@ -16,7 +16,7 @@ class Graph: self.build(root) self.draw() - def build(self, x: Value): + def build(self, x: Scalar): self.nodes.add(x) for child in x._prev: # Add a line from child to x diff --git a/src/value.py b/src/scalar.py index af0ab35..a67b7ae 100644 --- a/src/value.py +++ b/src/scalar.py @@ -1,4 +1,4 @@ -class Value: +class Scalar: def __init__(self, data, _children=(), _op='', label='') -> None: self.label = label @@ -9,12 +9,12 @@ class Value: self._op = _op def __repr__(self) -> str: - return f'Value({self.data})' + return f'Scalar({self.data})' def __add__(self, y): result = self.data + y.data - return Value(result, (self, y), _op='+') + return Scalar(result, (self, y), _op='+') def __mul__(self, y): result = self.data * y.data - return Value(result, (self, y), _op='*') + return Scalar(result, (self, y), _op='*') |