Method split

client/src/featureform/serving.py:579–615

(self)

                yield request

    def split(self):
        response_stream = self.stub.TrainingTestSplit(self.request_generator())
        # verify initialization response # TODO: don't think we need this
        init_response = next(response_stream)
        if not init_response.initialized:
            raise ValueError("Failed to initialize training test split")

        self.train_iter = TrainingSetSplitIterator(
            req_queue=self.req_queue,
            resp_queue=self.resp_queue,
            resp_stream=response_stream,
            request_type=serving_pb2.RequestType.TRAINING,
            name=self.name,
            version=self.version,
            model=self.model,
            test_size=self.test_size,
            train_size=self.train_size,
            shuffle=self.shuffle,
            random_state=self.random_state,
            batch_size=self.batch_size,
        )
        self.test_iter = TrainingSetSplitIterator(
            req_queue=self.req_queue,
            resp_queue=self.resp_queue,
            resp_stream=response_stream,
            request_type=serving_pb2.RequestType.TEST,
            name=self.name,
            version=self.version,
            model=self.model,
            test_size=self.test_size,
            train_size=self.train_size,
            shuffle=self.shuffle,
            random_state=self.random_state,
            batch_size=self.batch_size,
        )

        return self.train_iter, self.test_iter


class Dataset: