Update README.

Elvenson · Oct 31, 2020 · 10f0703 · 10f0703
1 parent 7b13e0c
commit 10f0703
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/README.md b/README.md
@@ -31,13 +31,13 @@ package main
 import (
 	"fmt"
 
+	xgb "github.com/Elvenson/xgboost-go"
 	"github.com/Elvenson/xgboost-go/activation"
 	"github.com/Elvenson/xgboost-go/mat"
-	"github.com/Elvenson/xgboost-go/models"
 )
 
 func main() {
-	ensemble, err := models.LoadXGBoostFromJSON("your model path",
+	ensemble, err := xgb.LoadXGBoostFromJSON("your model path",
 		"", 1, 4, &activation.Logistic{})
 	if err != nil {
 		panic(err)