Skip to content

A
alpha-mind
Commit 2cf8fae1 authored by Dr.李's avatar Dr.李
Browse files
Options

fixed bug

parent 5ce84621
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
alphamind/model/modelbase.py
View file @ 2cf8fae1
...@@ -22,7 +22,7 @@ class ModelBase(metaclass=abc.ABCMeta): ...@@ -22,7 +22,7 @@ class ModelBase(metaclass=abc.ABCMeta):
self.trained_time = None self.trained_time = None
def fit(self, x, y): def fit(self, x, y):
self.impl.fit(x, x.flatten()) self.impl.fit(x, y.flatten())
self.trained_time = arrow.now().format("YYYY-MM-DD HH:mm:ss") self.trained_time = arrow.now().format("YYYY-MM-DD HH:mm:ss")
def predict(self, x: np.ndarray) -> np.ndarray: def predict(self, x: np.ndarray) -> np.ndarray:
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment