Skip to content

Commit cb774c3

Browse files
committed
MiniBatchFraction added in related PythonMLLibAPI java stubs.
1 parent 81fcbc6 commit cb774c3

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

mllib/src/main/scala/org/apache/spark/mllib/api/python/PythonMLLibAPI.scala

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -271,6 +271,7 @@ class PythonMLLibAPI extends Serializable {
271271
.setNumIterations(numIterations)
272272
.setRegParam(regParam)
273273
.setStepSize(stepSize)
274+
.setMiniBatchFraction(miniBatchFraction)
274275
if (regType == "l2") {
275276
lrAlg.optimizer.setUpdater(new SquaredL2Updater)
276277
} else if (regType == "l1") {
@@ -350,6 +351,7 @@ class PythonMLLibAPI extends Serializable {
350351
.setNumIterations(numIterations)
351352
.setRegParam(regParam)
352353
.setStepSize(stepSize)
354+
.setMiniBatchFraction(miniBatchFraction)
353355
if (regType == "l2") {
354356
SVMAlg.optimizer.setUpdater(new SquaredL2Updater)
355357
} else if (regType == "l1") {
@@ -383,6 +385,7 @@ class PythonMLLibAPI extends Serializable {
383385
.setNumIterations(numIterations)
384386
.setRegParam(regParam)
385387
.setStepSize(stepSize)
388+
.setMiniBatchFraction(miniBatchFraction)
386389
if (regType == "l2") {
387390
LogRegAlg.optimizer.setUpdater(new SquaredL2Updater)
388391
} else if (regType == "l1") {

0 commit comments

Comments
 (0)