How to use the model
#1
by
aarbues
- opened
Since it's not written int the model card, here is how to use this model for evaluation.
import pickle
model_path = load_from_hub(repo_id="VcRlAgent/VC-Taxi-v3-Gymnasium", filename="q-learning.pkl")
model_obj = pickle.load(open(model_path, "rb"))
qtable = model_obj['qtable']
def get_q_action(obs):
return np.argmax(qtable[obs])