Yes I was thinking of cloning the repo but I am afraid my laptop might not have the required specs or might hang during the training. I am eagerly waiting for codespaces on GitHub for that. I shall try to reduce the training set size. Could you figure out from the snippet posted above about the shape of the output variable?