from tfx.components import Trainer
from tfx.proto import trainer_pb2

trainer = Trainer(
    module_file='/content/trainer_module.py',
    examples=transform.outputs['transformed_examples'],
    schema=schema_gen.outputs['schema'],
    transform_graph=transform.outputs['transform_graph'],
    train_args=trainer_pb2.TrainArgs(num_steps=1000),
    eval_args=trainer_pb2.EvalArgs(num_steps=100)
)

        __
        __