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