Skip to content

Commit a7576b4

Browse files
author
Dan Choi
committed
Fix rcf default batch_size
1 parent 5df7840 commit a7576b4

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/sagemaker/amazon/randomcutforest.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -87,7 +87,7 @@ def create_model(self):
8787

8888
return RandomCutForestModel(self.model_data, self.role, sagemaker_session=self.sagemaker_session)
8989

90-
def _prepare_for_training(self, records, mini_batch_size=MINI_BATCH_SIZE, job_name=None):
90+
def _prepare_for_training(self, records, mini_batch_size=1000, job_name=None):
9191
if mini_batch_size != self.MINI_BATCH_SIZE:
9292
raise ValueError("Random Cut Forest uses a fixed mini_batch_size of {}"
9393
.format(self.MINI_BATCH_SIZE))

0 commit comments

Comments
 (0)