File tree Expand file tree Collapse file tree 1 file changed +3
-0
lines changed
mllib/src/main/scala/org/apache/spark/mllib/api/python Expand file tree Collapse file tree 1 file changed +3
-0
lines changed Original file line number Diff line number Diff line change @@ -271,6 +271,7 @@ class PythonMLLibAPI extends Serializable {
271
271
.setNumIterations(numIterations)
272
272
.setRegParam(regParam)
273
273
.setStepSize(stepSize)
274
+ .setMiniBatchFraction(miniBatchFraction)
274
275
if (regType == " l2" ) {
275
276
lrAlg.optimizer.setUpdater(new SquaredL2Updater )
276
277
} else if (regType == " l1" ) {
@@ -350,6 +351,7 @@ class PythonMLLibAPI extends Serializable {
350
351
.setNumIterations(numIterations)
351
352
.setRegParam(regParam)
352
353
.setStepSize(stepSize)
354
+ .setMiniBatchFraction(miniBatchFraction)
353
355
if (regType == " l2" ) {
354
356
SVMAlg .optimizer.setUpdater(new SquaredL2Updater )
355
357
} else if (regType == " l1" ) {
@@ -383,6 +385,7 @@ class PythonMLLibAPI extends Serializable {
383
385
.setNumIterations(numIterations)
384
386
.setRegParam(regParam)
385
387
.setStepSize(stepSize)
388
+ .setMiniBatchFraction(miniBatchFraction)
386
389
if (regType == " l2" ) {
387
390
LogRegAlg .optimizer.setUpdater(new SquaredL2Updater )
388
391
} else if (regType == " l1" ) {
You can’t perform that action at this time.
0 commit comments