search

tensorflow / agents

TF-Agents: A reliable, scalable and easy to use TensorFlow library for Contextual Bandits and Reinforcement Learning.
Apache License 2.0
2.8k stars 721 forks source link

Inference MAB #914

Open Akshaysharma29 opened 9 months ago

Akshaysharma29 commented 9 months ago

Hi Team, Thanks for creating such wonderful notebooks and examples. 😊 I have successfully trained a model(For movie lens data) using a training script mentioned in the example section(https://github.com/tensorflow/agents/blob/master/tf_agents/bandits/agents/examples/v2/train_eval_movielens.py)

Can you guide me on how can I use the trained model to infer for a particular user(I want to deploy it)?

Akshaysharma29 commented 9 months ago

Hi Team, just adding the cmd which I used: python train_eval_movielens.py --data_path /content/sample_data --root_dir /content/tmp --agent epsGreedy --per_arm False