We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent fed16df commit 343a923Copy full SHA for 343a923
1 file changed
python/runtime/xgboost/evaluate.py
@@ -15,6 +15,7 @@
15
import sklearn.metrics
16
import xgboost as xgb
17
from runtime import db
18
+from runtime.dbapi.paiio import PaiIOConnection
19
from runtime.xgboost.dataset import xgb_dataset
20
21
SKLEARN_METRICS = [
@@ -58,7 +59,7 @@ def evaluate(datasource,
58
59
if not is_pai:
60
conn = db.connect_with_data_source(datasource)
61
else:
- conn = None
62
+ conn = PaiIOConnection.from_table(pai_table)
63
dpred = xgb_dataset(datasource,
64
'predict.txt',
65
select,
0 commit comments