Method ExecuteExplain

go/executor/executor.go:265–307

(cl *ir.ExplainStmt)

}

func (s *pythonExecutor) ExecuteExplain(cl *ir.ExplainStmt) error {
	// NOTE(typhoonzero): model is already loaded under s.Cwd
	var code string
	var err error
	db, err := database.OpenAndConnectDB(s.Session.DbConnStr)
	if err != nil {
		return err
	}
	defer db.Close()

	var modelType int
	if cl.TrainStmt.GetModelKind() == ir.XGBoost {
		code, err = xgboost.Explain(cl, s.Session)
		modelType = model.XGBOOST
	} else {
		code, err = tensorflow.Explain(cl, s.Session)
		modelType = model.TENSORFLOW
	}

	if cl.Into != "" {
		err := createExplainResultTable(db, cl, cl.Into, modelType, cl.TrainStmt.Estimator)
		if err != nil {
			return err
		}
	}

	if err != nil {
		return err
	}
	if err = s.runProgram(code, false); err != nil {
		return err
	}
	img, err := readExplainResult(path.Join(s.Cwd, "summary.png"))
	if err != nil {
		return err
	}
	termFigure, err := ioutil.ReadFile(path.Join(s.Cwd, "summary.txt"))
	if err != nil {
		return err
	}
	s.Writer.Write(Figures{img, string(termFigure)})
	return nil
}

func (s *pythonExecutor) ExecuteEvaluate(cl *ir.EvaluateStmt) error {
	// NOTE(typhoonzero): model is already loaded under s.Cwd