Movie rating prediction

year duration nominations categories rating
1921 3240 0 comedy drama 8.4
1925 5700 1 adventure comedy 8.3
1927 9180 4 drama comedy scifi 8.4
ludwig experiment \
--dataset movie_ratings.csv \
  --config_file config.yaml

With config.yaml:

input_features:
    -
        name: year
        type: numerical
    -
        name: duration
        type: numerical
    -
        name: nominations
        type: numerical
    -
        name: categories
        type: set

output_features:
    -
        name: rating
        type: numerical