// GetModelKind returns the kind of model in the TrainStmt
GetModelKind returns the kind of model in the TrainStmt
// GetModelKind returns the kind of model in the TrainStmt
func (cl *TrainStmt) GetModelKind() int {
	estimator := strings.ToUpper(cl.Estimator)
	if strings.HasPrefix(estimator, "XGB") {
		return XGBoost
	}
	if strings.HasPrefix(estimator, "KMeans") {
		return KMeans
	}
	return TensorFlow
}
func ( cl * TrainStmt ) GetModelKind ( ) int { estimator := strings . ToUpper ( cl . Estimator ) if strings . HasPrefix ( estimator , " " ) { return XGBoost } if strings . HasPrefix ( estimator , " " ) { return KMeans } return TensorFlow }
GetModelKind returns the kind of model in the TrainStmt
GetModelKind returns the kind of model in the TrainStmt