nasa / concept-tagging-api

Contains code for the API that takes in text and predicts concepts & keywords from a list of standardized NASA keywords. API is for exposing models created with the repository `concept-tagging-training`.
MIT License
19 stars 10 forks source link

Off-platform question on why this uses Scikit-learn vs. TensorFlow & PyTorch? #7

Closed JustinGOSSES closed 3 years ago

JustinGOSSES commented 3 years ago

There was an off-platform question about why this uses Scikit-learn instead of Tensorflow or pyTorch.

Sort answer is there is no reason this couldn't be ported from Scikit-learn to Tensorflow or PyTorch. There just wasn't a big enough driver to port it. The speed internally was sufficient as is.