class Config(NewDocModel.Config):
class RegressionModelInput(Model.Config.ModelInput):
tokens: TokenTensorizer.Config = TokenTensorizer.Config()
labels: NumericLabelTensorizer.Config = NumericLabelTensorizer.Config()
inputs: RegressionModelInput = RegressionModelInput()
output_layer: RegressionOutputLayer.Config = RegressionOutputLayer.Config()
class Config(NewBertModel.Config):
class InputConfig(ConfigBase):
tokens: BERTTensorizer.Config = BERTTensorizer.Config(
columns=["text1", "text2"], max_seq_len=128
)
labels: NumericLabelTensorizer.Config = NumericLabelTensorizer.Config()
inputs: InputConfig = InputConfig()
output_layer: RegressionOutputLayer.Config = RegressionOutputLayer.Config()
