Skip to content

Commit

Permalink
Merge pull request #16 from NvTimLiu/release-1.4.2-0.3.0
Browse files Browse the repository at this point in the history
Merge branch 'GitLab/nv-release-1.4.0' into nv-release_1.4.0
  • Loading branch information
GaryShen2008 authored Apr 24, 2022
2 parents 983159f + c5927a2 commit a79c7e4
Show file tree
Hide file tree
Showing 5 changed files with 28 additions and 9 deletions.
14 changes: 14 additions & 0 deletions SECURITY.md
Original file line number Diff line number Diff line change
@@ -0,0 +1,14 @@
## Security

NVIDIA is dedicated to the security and trust of our software products and services, including all source code repositories managed through our organization.

If you need to report a security issue, please use the appropriate contact points outlined below. **Please do not report security vulnerabilities through GitHub/GitLab.**

## Reporting Potential Security Vulnerability in an NVIDIA Product

To report a potential security vulnerability in any NVIDIA product:
- Web: [Security Vulnerability Submission Form](https://www.nvidia.com/object/submit-security-vulnerability.html)
- E-Mail: [email protected]
- We encourage you to use the following PGP key for secure email communication: [NVIDIA public PGP Key for communication](https://www.nvidia.com/en-us/security/pgp-key)
- Please include the following information:
- Product/Driver name and version/branch that contains the vulnerability
2 changes: 1 addition & 1 deletion jvm-packages/pom.xml
Original file line number Diff line number Diff line change
Expand Up @@ -6,7 +6,7 @@

<groupId>com.nvidia</groupId>
<artifactId>xgboost-jvm_3.0</artifactId>
<version>1.4.2-0.2.0</version>
<version>1.4.2-0.3.0</version>
<packaging>pom</packaging>
<name>XGBoost JVM Package</name>
<description>JVM Package for XGBoost</description>
Expand Down
4 changes: 2 additions & 2 deletions jvm-packages/xgboost4j-spark/pom.xml
Original file line number Diff line number Diff line change
Expand Up @@ -6,7 +6,7 @@
<parent>
<groupId>com.nvidia</groupId>
<artifactId>xgboost-jvm_3.0</artifactId>
<version>1.4.2-0.2.0</version>
<version>1.4.2-0.3.0</version>
</parent>
<artifactId>xgboost4j-spark_3.0</artifactId>
<build>
Expand All @@ -24,7 +24,7 @@
<dependency>
<groupId>com.nvidia</groupId>
<artifactId>xgboost4j_3.0</artifactId>
<version>1.4.2-0.2.0</version>
<version>1.4.2-0.3.0</version>
</dependency>
<dependency>
<groupId>org.apache.spark</groupId>
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -26,10 +26,15 @@ def __init__(self):
'ml.dmlc.xgboost4j.scala.spark.rapids.CrossValidator')

def fit(self, dataset):
java_estimator, java_epms, java_evaluator = self._to_java_impl()
self._java_obj.setEstimator(java_estimator)
self._java_obj.setEvaluator(java_evaluator)
self._java_obj.setEstimatorParamMaps(java_epms)
estimator, epms, evaluator = self._to_java_impl()

self._java_obj.setEstimatorParamMaps(epms)
self._java_obj.setEvaluator(evaluator)
self._java_obj.setEstimator(estimator)
self._java_obj.setSeed(self.getSeed())
self._java_obj.setNumFolds(self.getNumFolds())
self._java_obj.setParallelism(self.getParallelism())
self._java_obj.setCollectSubModels(self.getCollectSubModels())

cv_java_model = self._java_obj.fit(dataset._jdf)
cv_py_model = CrossValidatorModel._from_java(cv_java_model)
Expand Down
4 changes: 2 additions & 2 deletions jvm-packages/xgboost4j/pom.xml
Original file line number Diff line number Diff line change
Expand Up @@ -6,10 +6,10 @@
<parent>
<groupId>com.nvidia</groupId>
<artifactId>xgboost-jvm_3.0</artifactId>
<version>1.4.2-0.2.0</version>
<version>1.4.2-0.3.0</version>
</parent>
<artifactId>xgboost4j_3.0</artifactId>
<version>1.4.2-0.2.0</version>
<version>1.4.2-0.3.0</version>
<packaging>jar</packaging>

<dependencies>
Expand Down

0 comments on commit a79c7e4

Please sign in to comment.