Skip to content

Commit e9351cc

Browse files
authored
Fix kmeans name check (#2753)
* refine pai code structure * remove unused function * move tensorflow files to a sub-package * add docstring * Add KMeans and RF PAI submitter * use model type defined in runtim.model * fix string compare * Add log for model zoo server
1 parent f0f1d30 commit e9351cc

File tree

2 files changed

+2
-1
lines changed

2 files changed

+2
-1
lines changed

go/ir/ir.go

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -80,7 +80,7 @@ func (stmt *TrainStmt) GetModelKind() int {
8080
if strings.HasPrefix(estimator, "XGB") {
8181
return XGBoost
8282
}
83-
if strings.HasPrefix(estimator, "KMeans") {
83+
if strings.HasPrefix(estimator, "KMEANS") {
8484
return KMeans
8585
}
8686
return TensorFlow

go/modelzooserver/modelzooserver.go

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -98,6 +98,7 @@ func StartModelZooServer(port int, dbConnStr string) {
9898

9999
pb.RegisterModelZooServerServer(grpcServer, &modelZooServer{DB: mysqlConn})
100100

101+
logger.Infof("SQLFlow Model Zoo started at: %d", port)
101102
grpcServer.Serve(lis)
102103
}
103104

0 commit comments

Comments
 (0)