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])

Sign up or log in to comment