DNNClassifer

• 训练

%%sqlflow
SELECT * FROM iris.train TO TRAIN DNNClassifier WITH
  # 分类的数量
  model.n_classes = 3,
  # 隐藏单元个数
  model.hidden_units = [10, 10],
  # 优化器的学习率，DNNClassifier的默认优化器是AdaGrad，默认学习率为0.001
  optimizer.learning_rate=0.1,
  # 定义epoch，1个epoch是指把所有训练数据跑一遍
  train.epoch = 10
# 特征列
COLUMN sepal_length, sepal_width, petal_length, petal_width
# 标签列
LABEL class
# 保存模型
INTO sqlflow_models.my_dnn_model;

• 自动训练

%%sqlflow
SELECT * FROM iris.train TO TRAIN sqlflow_models.AutoClassifier WITH
  model.n_classes = 3,
  train.epoch = 10
COLUMN sepal_length, sepal_width, petal_length, petal_width
LABEL class
INTO sqlflow_models.my_dnn_model;

• 预测

%%sqlflow
SELECT * FROM iris.test TO PREDICT iris.predict.class USING sqlflow_models.my_dnn_model;