Skip to content

Commit 5018182

Browse files
rey-esptswast
andauthored
Update bigframes/ml/decomposition.py
Co-authored-by: Tim Sweña (Swast) <[email protected]>
1 parent e0ef53e commit 5018182

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

bigframes/ml/decomposition.py

+5
Original file line numberDiff line numberDiff line change
@@ -271,6 +271,11 @@ def __init__(
271271
self._bqml_model: Optional[core.BqmlModel] = None
272272
self._bqml_model_factory = globals.bqml_model_factory()
273273

274+
@property
275+
def rating_col(self) -> str:
276+
"""str: The rating column name. Defaults to 'rating'.""""
277+
return self._input_label_columns[0]
278+
274279
@classmethod
275280
def _from_bq(
276281
cls, session: bigframes.session.Session, bq_model: bigquery.Model

0 commit comments

Comments
 (0)