Custom top-N model

Solved!
Erlebacher
Level 4
Custom top-N model

I am having issues with my custom model. From what I read, models should derive from `scikit-learn` and I should make sure I have a `classes_` attribute in my model. However, my model is a top-N recommender, and I am using `rankfm`, a factorization machine model available in Python/C on github. I have included all the relevant modules in my environment. The choices I have are related to clustering, classification, and regression, but I am doing none of these. How do I create my custom model so that it can be integrated into a Dataiku workflow? I want my prediction function to output a top-N recommendation in a DataFrame. I have gone through various tutorials, but found none that address my use case. Any help is appreciated. Thanks. 

     Gordon


Operating system used: Mac Ventura

0 Kudos
1 Solution
HarizoR
Developer Advocate

Hi,

This is handled in the MLFlow part where you save your trained MLFlow model as a Dataiku Saved Model version: https://doc.dataiku.com/dss/latest/mlops/mlflow-models/importing.html

View solution in original post

0 Kudos
8 Replies
HarizoR
Developer Advocate

Hi,

Recommender systems are a specific class of ML problems that indeed do not fit in the usual classification/regression/clustering canvas. The simplest alternative would be to write your code directly either in a Python recipe or in a notebook.

Best,

 

Harizo

0 Kudos
Erlebacher
Level 4
Author

I have been able to run my algorithm inside a multiclass custom code. However, my question is this: how to I hook up an output dataset to the custom code? More specifically, I wish to write a predict() function with my own output format. I want to be able to avail myself of Dataiku's evaluation tools and have my prediction flow downstream in the flow. Thanks.

0 Kudos
Erlebacher
Level 4
Author

Hi @HarizoR . I have written my code in a recipe and notebook. The reason I tried the custom model is that my model has a fit() and predict() stage. The fit stage reads training data, and the predict stage reads in a validation file and produces a new data frame. How do I combine fitting and prediction stages when I use my own recipes? Are there any resources on that? 

Finally, is it possible to use the various tools of Dataiku (AUC, precision, etc.) if I use my full recipes, bypassing the custom model? I have not yet come across an existing model of that type. Any hints would be appreciated. 

0 Kudos
HarizoR
Developer Advocate

Hi,

Custom models are usually a good fit when the architecture of your code only varies slightly compared to a standard scikit-learn estimator. In your case you need more flexibility hence the "custom" alternative using Python recipes/notebooks.

Since Dataiku 11, you can train custom complex models entirely using code and leverage the built-in model result visualizations by leveraging the new experiment tracking and MLFlow model import features. By turning your code into something MLFlow-compatible (see example here) you should be able to retrieve the insights you are looking for.

Best,

Harizo

0 Kudos
Erlebacher
Level 4
Author

Thanks. That is how I stayed. But I could not figure out how to save the model after training, so I could later generate a score. Has that been addressed? People told me earlier that I needed a custom algorithm. Thanks.

0 Kudos
HarizoR
Developer Advocate

Hi,

This is handled in the MLFlow part where you save your trained MLFlow model as a Dataiku Saved Model version: https://doc.dataiku.com/dss/latest/mlops/mlflow-models/importing.html

0 Kudos
Erlebacher
Level 4
Author

Hi @HarizoR ,

Your solution makes sense. In the docs, I read:

This section focuses on the deployment through the API. It assumes that you already have a MLflow model in a model_directory, i.e. a local folder on the local filesystem, or a Managed Folder.

So I am back to my original problem. How do I get an MLflow model into a model directory. My original question had been how to save a model, and nobody could answer me at the time, although I was already using Dataiku 11.

Is MLFlow (experimental) available on the free version of Dataiku? Thanks.

0 Kudos
Erlebacher
Level 4
Author

I think I understand. I train the model on my laptop, save it with pickle, and use the MLFlow API (assuming it is available to my in the free version of Dataiku 11.x) to uploaded it to Dataiku. If so, it makes sense.
I noticed your use of cloudpickle (which I had not heard about).

Cheers,

Gordon

0 Kudos