Skip to content

Instantly share code, notes, and snippets.

Embed
What would you like to do?
Example Ludwig modeldef file
training:
batch_size: 32
epochs: 400
early_stop: 50
learning_rate: 0.001
optimizer:
type: rmsprop
input_features:
-
name: Cylinders
type: numerical
-
name: Displacement
type: numerical
-
name: Horsepower
type: numerical
-
name: Weight
type: numerical
-
name: Acceleration
type: numerical
-
name: ModelYear
type: numerical
-
name: Origin
type: category
output_features:
-
name: MPG
type: numerical
optimizer:
type: mean_squared_error
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
You can’t perform that action at this time.