蚂蚁金服开源 SQLFlow,训练模型像 SQL 一样简单。

2019-05-06  本文已影响0人  一根筋网络

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; --选择已训练好的模型

上一篇 下一篇

猜你喜欢

热点阅读