File tree 2 files changed +2
-9
lines changed
examples/XGBoost-Examples
utility/scala/src/com/nvidia/spark/examples/utility
2 files changed +2
-9
lines changed Original file line number Diff line number Diff line change 38
38
39
39
<properties >
40
40
<encoding >UTF-8</encoding >
41
- <xgboost .version>2.1 .0-SNAPSHOT</xgboost .version>
41
+ <xgboost .version>2.2 .0-SNAPSHOT</xgboost .version>
42
42
<spark .version>3.5.0</spark .version>
43
43
<scala .version>2.12.8</scala .version>
44
44
<scala .binary.version>2.12</scala .binary.version>
45
45
</properties >
46
46
47
47
<dependencies >
48
- <dependency >
49
- <groupId >ml.dmlc</groupId >
50
- <artifactId >xgboost4j-gpu_${scala.binary.version}</artifactId >
51
- <version >${xgboost.version} </version >
52
- </dependency >
53
48
<dependency >
54
49
<groupId >ml.dmlc</groupId >
55
50
<artifactId >xgboost4j-spark-gpu_${scala.binary.version}</artifactId >
Original file line number Diff line number Diff line change @@ -22,8 +22,6 @@ import com.google.common.base.CaseFormat
22
22
import scala .collection .mutable
23
23
import scala .util .Try
24
24
25
- import ml .dmlc .xgboost4j .scala .spark .TrackerConf
26
-
27
25
private case class XGBoostArg (
28
26
required : Boolean = false ,
29
27
parse : String => Any = value => value,
@@ -202,7 +200,7 @@ class XGBoostArgs private[utility] (
202
200
203
201
val hostIp = params.getOrElse(" rabit_tracker_host" , " " ).toString
204
202
if (! hostIp.isEmpty) {
205
- params ++ Map (" tracker_conf " -> TrackerConf ( 0 , hostIp) )
203
+ params ++ Map (" rabitTrackerHostIp " -> hostIp)
206
204
} else params
207
205
}
208
206
You can’t perform that action at this time.
0 commit comments