diff --git a/nada_ai/linear_model/linear_regression.py b/nada_ai/linear_model/linear_regression.py index eead976..31f2764 100644 --- a/nada_ai/linear_model/linear_regression.py +++ b/nada_ai/linear_model/linear_regression.py @@ -4,7 +4,7 @@ from nada_ai.nn.module import Module from nada_ai.nn.parameter import Parameter -__all__ = ["LinearRegression"] +__all__ = ["LinearRegression", "LogisticRegression"] class LinearRegression(Module):