Skip to content

feat: add bigframes.bigquery.ml methods #2300

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
tswast merged 27 commits into from
Dec 3, 2025
Merged

feat: add bigframes.bigquery.ml methods #2300

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
27 commits
Select commit Hold shift + click to select a range
0356206
feat: Add BigQuery ML CREATE MODEL support
google-labs-jules[bot] Dec 1, 2025
54050d5
feat: Add BigQuery ML CREATE MODEL support
google-labs-jules[bot] Dec 1, 2025
fdee53f
feat: Add BigQuery ML CREATE MODEL support
google-labs-jules[bot] Dec 1, 2025
c1adfd9
feat: Add BigQuery ML CREATE MODEL support
google-labs-jules[bot] Dec 1, 2025
8b08869
feat: Add BigQuery ML CREATE MODEL support
google-labs-jules[bot] Dec 1, 2025
bfc4fdf
fix lint errors
tswast Dec 2, 2025
15d1d59
update docs
tswast Dec 2, 2025
261073d
revert sample
tswast Dec 2, 2025
50e98ff
Merge branch 'main' into create-model-support
tswast Dec 2, 2025
b809f81
feat: Add BigQuery ML CREATE MODEL support
google-labs-jules[bot] Dec 2, 2025
7df5e09
Revert "feat: Add BigQuery ML CREATE MODEL support"
tswast Dec 2, 2025
9231f56
restore docstring from jules
tswast Dec 2, 2025
f323b6b
link to sql reference
tswast Dec 2, 2025
2d5e065
add todo for other transform inputs
tswast Dec 2, 2025
c86e15a
create bbq notebook
tswast Dec 2, 2025
9ad9011
add more functions
tswast Dec 2, 2025
b355e47
add more functions
tswast Dec 2, 2025
b4e31ef
fix struct options
tswast Dec 2, 2025
a59d746
add sample notebook
tswast Dec 2, 2025
fba9326
feat: Add BigQuery ML CREATE MODEL support
google-labs-jules[bot] Dec 3, 2025
74d4fcc
Revert "feat: Add BigQuery ML CREATE MODEL support"
tswast Dec 3, 2025
68e770b
return pd.Series from `create_model`
tswast Dec 3, 2025
82d1aec
support pandas inputs
tswast Dec 3, 2025
1d88694
add unit tests
tswast Dec 3, 2025
84f4427
add less mocking
tswast Dec 3, 2025
e43636b
Merge remote-tracking branch 'origin/main' into create-model-support
tswast Dec 3, 2025
9baeb8a
skip snapshot tests where not present
tswast Dec 3, 2025
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
3 changes: 2 additions & 1 deletion bigframes/bigquery/__init__.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -18,7 +18,7 @@

import sys

from bigframes.bigquery import ai
from bigframes.bigquery import ai, ml
from bigframes.bigquery._operations.approx_agg import approx_top_count
from bigframes.bigquery._operations.array import (
array_agg,
Expand Down Expand Up @@ -157,4 +157,5 @@
"struct",
# Modules / SQL namespaces
"ai",
"ml",
]
Loading