Method ExecutePredict

go/executor/pai.go:300–342

(cl *ir.PredictStmt)

}

func (s *paiExecutor) ExecutePredict(cl *ir.PredictStmt) error {
	// TODO(typhoonzero): Do **NOT** create tmp table when the select statement is like:
	// "SELECT fields,... FROM table"
	dbName, tableName, err := createTmpTableFromSelect(cl.Select, s.Session.DbConnStr)
	if err != nil {
		return err
	}
	cl.TmpPredictTable = strings.Join([]string{dbName, tableName}, ".")
	defer dropTmpTables([]string{cl.TmpPredictTable}, s.Session.DbConnStr)

	currProject, err := database.GetDatabaseName(s.Session.DbConnStr)
	if err != nil {
		return err
	}
	// format resultTable name to "db.table" to let the codegen form a submitting
	// argument of format "odps://project/tables/table_name"
	resultTableParts := strings.Split(cl.ResultTable, ".")
	if len(resultTableParts) == 1 {
		cl.ResultTable = fmt.Sprintf("%s.%s", currProject, cl.ResultTable)
	}
	if e := createPredictionResultTable(cl, s.Db, s.Session); e != nil {
		return e
	}

	ossModelPath, e := getModelPath(cl.Using, s.Session)
	if e != nil {
		return e
	}
	modelType, estimator, err := getOSSSavedModelType(ossModelPath, currProject)
	if err != nil {
		return err
	}
	scriptPath := fmt.Sprintf("file://%s/%s", s.Cwd, tarball)
	paramsPath := fmt.Sprintf("file://%s/%s", s.Cwd, paramsFile)
	if err := createPAIHyperParamFile(s.Cwd, paramsFile, ossModelPath); err != nil {
		return err
	}
	code, paiCmd, requirements, e := pai.Predict(cl, s.Session, scriptPath, paramsPath, cl.Using, ossModelPath, s.Cwd, modelType)
	if e != nil {
		return e
	}
	return s.submitPAITask(code, paiCmd, requirements, estimator)
}

func (s *paiExecutor) ExecuteExplain(cl *ir.ExplainStmt) error {
	// TODO(typhoonzero): Do **NOT** create tmp table when the select statement is like: