Leson 3 Rossman - Can I use an image as a continuous variable?

I’d like to include images as features for my model, the dataset I have contains images of the product that made the sale and it would be interesting that the image contributed to the prediction of the sales, since design is an important decision factor for the products being sold on this dataset.

Can those images being used on the model somehow?

Maybe you can use a pretrained ResNet34 and use the last activations as additional inputs to the MixedInputModel.

1 Like

You could also create an image classifier for predicting the class of the image, feed in the predicted class as a new column in your overall training set, and train from there.