Skip to content

Instantly share code, notes, and snippets.

@hamelsmu

hamelsmu/vectorize_code.py

Last active May 27, 2018 04:49
Show Gist options
  • Save hamelsmu/8b7144f6ec0f6c9a5c462e95adb944e2 to your computer and use it in GitHub Desktop.
Save hamelsmu/8b7144f6ec0f6c9a5c462e95adb944e2 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
vectorize_code.py
# transform raw docstring input
encinp = enc_pp.transform_parallel(no_docstring_funcs)
# vectorize code using the code2emb model
nodoc_vecs = code2emb_model.predict(encinp, batch_size=20000)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment