Skip to content

Commit a915e16

Browse files
committed
LogisticRegressionOperator with Python API support
1 parent fecfa4f commit a915e16

1 file changed

Lines changed: 5 additions & 1 deletion

File tree

python/src/pywy/dataquanta.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -195,7 +195,11 @@ def predict(
195195
)
196196

197197

198-
def train_logistic_regression(self, labels, fit_intercept=True):
198+
def train_logistic_regression(
199+
self: "DataQuanta[In]",
200+
labels: "DataQuanta[In]",
201+
fit_intercept: bool = True
202+
) -> "DataQuanta[Out]":
199203
from pywy.basic.model.ops import LogisticRegression
200204

201205
op = LogisticRegression()

0 commit comments

Comments
 (0)