experimental

Documentation

Index

• func CodeGenCouler(stepList []*stepContext, session *pb.Session) (string, error)
• func GenerateCodeCouler(sqlProgram string, session *pb.Session) (string, error)
• func GeneratePyDbConnStr(session *pb.Session) (string, error)
• func InitializeAttributes(trainStmt *ir.TrainStmt) error
• func XGBoostGenerateEvaluation(evalStmt *ir.EvaluateStmt, stepIndex int, session *pb.Session) (string, error)
• func XGBoostGeneratePredict(predStmt *ir.PredictStmt, stepIndex int, session *pb.Session) (string, error)
• func XGBoostGenerateTrain(trainStmt *ir.TrainStmt, stepIndex int, session *pb.Session) (string, error)

Functions

func CodeGenCouler

func CodeGenCouler(stepList []*stepContext, session *pb.Session) (string, error)

CodeGenCouler generate couler code to generate a workflow

func GenerateCodeCouler

func GenerateCodeCouler(sqlProgram string, session *pb.Session) (string, error)

GenerateCodeCouler generate a Couler program to submit a workflow to run the sql program. 1. generate IR of each statement. 2. generate runtime code of each statement 3. generate couler program to form a workflow

func GeneratePyDbConnStr

func GeneratePyDbConnStr(session *pb.Session) (string, error)

GeneratePyDbConnStr generates the db connection string for the Python dbapi.

func InitializeAttributes

func InitializeAttributes(trainStmt *ir.TrainStmt) error

InitializeAttributes initializes the attributes of XGBoost and does type checking for them

func XGBoostGenerateEvaluation

func XGBoostGenerateEvaluation(evalStmt *ir.EvaluateStmt, stepIndex int, session *pb.Session) (string, error)

XGBoostGenerateEvaluation generates the XGBoost evaluation code

func XGBoostGeneratePredict

func XGBoostGeneratePredict(predStmt *ir.PredictStmt, stepIndex int, session *pb.Session) (string, error)

XGBoostGeneratePredict generates the XGBoost prediction code

func XGBoostGenerateTrain

func XGBoostGenerateTrain(trainStmt *ir.TrainStmt, stepIndex int, session *pb.Session) (string, error)

XGBoostGenerateTrain returns the step code.