SQLFlow > https://github.com/sql-machine-learning/sqlflow
训练模型
SELECT *
FROM iris.train --从元数据表读数据
TRAIN DNNClassifier --选择 DNN 分类器
WITH n_classes = 3, hidden_units = [10, 20] --设置分类数(训练数据是三分类)和 hidden layer
COLUMN sepal_length, sepal_width, petal_length, petal_width --选择加入训练的数据列
LABEL iris.train.class --选择标注的 label 列
INTO sqlflow_models.my_dnn_model; --保存模型
用训练好的模型预测一组数据的 label
SELECT *
FROM iris.test --从待预测的数据表读数据( sepal_length, sepal_width, petal_length, petal_width 有数据,但 label 为空)
PREDICT iris.predict.class --待填充的 label 数据列
USING sqlflow_models.my_dnn_model; --选择已训练好的模型
网友评论