Use CREATE MODEL to train a model

eto-ai / rikai

Parquet-based ML data format optimized for working with unstructured data

https://rikai.readthedocs.io/en/latest/

Apache License 2.0

137 stars 19 forks source link

Use CREATE MODEL to train a model #163

Open eddyxu opened 3 years ago

eddyxu commented 3 years ago

Similar to what BigQuery ML offers, CREATE MODEL can be used for training a model as well.

CREATE MODEL yolo
OPTIONS (
     device="gpu",
     steps=10000,
     flavor="pytorch",
     model_type="yolov4",
)
FROM (SELECT id, image, objects FROM training_dataset)

da-liii commented 3 years ago

See https://cloud.google.com/bigquery-ml/docs/reference/standard-sql/bigqueryml-syntax-create