Browse code

Merge pull request #51 from SydneyBioX/master

XGBoost Fix Again

Dario Strbenac authored on 24/10/2022 23:32:01 • GitHub committed on 24/10/2022 23:32:01
Showing 2 changed files

... ...
@@ -3,8 +3,8 @@ Type: Package
3 3
 Title: A framework for cross-validated classification problems, with
4 4
        applications to differential variability and differential
5 5
        distribution testing
6
-Version: 3.1.27
7
-Date: 2022-10-24
6
+Version: 3.1.28
7
+Date: 2022-10-25
8 8
 Author: Dario Strbenac, Ellis Patrick, Sourish Iyengar, Harry Robertson, Andy Tran, John Ormerod, Graham Mann, Jean Yang
9 9
 Maintainer: Dario Strbenac <dario.strbenac@sydney.edu.au>
10 10
 VignetteBuilder: knitr
... ...
@@ -32,11 +32,10 @@ extremeGradientBoostingTrainInterface <- function(measurementsTrain, outcomeTrai
32 32
   }
33 33
 
34 34
   if(isClassification)
35
-  {
36 35
     attr(trained, "classes") <- classes # Useful for factor predictions in predict method.
37
-    attr(trained, "featureNames") <- colnames(measurementsTrain)
38
-    attr(trained, "featureGroups") <- measurementsTrain@assign
39
-  }
36
+  attr(trained, "featureNames") <- colnames(measurementsTrain)
37
+  attr(trained, "featureGroups") <- measurementsTrain@assign
38
+
40 39
   trained
41 40
 }
42 41
 attr(extremeGradientBoostingTrainInterface, "name") <- "extremeGradientBoostingTrainInterface"