Method ExecuteEvaluate

go/executor/executor.go:309–348

(cl *ir.EvaluateStmt)

}

func (s *pythonExecutor) ExecuteEvaluate(cl *ir.EvaluateStmt) error {
	// NOTE(typhoonzero): model is already loaded under s.Cwd
	var code string
	var err error
	if cl.TrainStmt.GetModelKind() == ir.XGBoost {
		code, err = xgboost.Evaluate(cl, s.Session)
		if err != nil {
			return err
		}
	} else {
		code, err = tensorflow.Evaluate(cl, s.Session)
		if err != nil {
			return err
		}
	}

	if cl.Into != "" {
		// create evaluation result table
		db, err := database.OpenAndConnectDB(s.Session.DbConnStr)
		if err != nil {
			return err
		}
		defer db.Close()
		// default always output evaluation loss
		metricNames := []string{"loss"}
		metricsAttr, ok := cl.Attributes["validation.metrics"]
		if ok {
			metricsList := strings.Split(metricsAttr.(string), ",")
			metricNames = append(metricNames, metricsList...)
		}
		err = createEvaluationResultTable(db, cl.Into, metricNames)
		if err != nil {
			return err
		}
	}
	if err = s.runProgram(code, false); err != nil {
		return err
	}
	return nil
}

func (s *pythonExecutor) ExecuteOptimize(stmt *ir.OptimizeStmt) error {
	db, err := database.OpenAndConnectDB(s.Session.DbConnStr)