Train machine learning model in Java Script

We need machine learning model to be trained on the data we provide. The model should give predictions of three numbers (low, mid & high).

Requirements:
1. Model should be trained with JavaScript framework (TensorFlow, Brain.js, etc., except no longer maintained frameworks);
2. It should be possible to load the model in Node.js runtime and get predictions;
3. Code written to train the model should be provided;
4. If the data is somehow transformed before training the model then it needs to be done with JS script. Basically we want to be able to train the model by ourselves in the future. Same applies to the data for getting the predictions;
5. Process of finding the features in data should be also scripted or clearly explained;
6. The solution should be documented;
7. Predictions should be accurate;
8. Getting predictions should be fast (500ms max);
9. Running model should not consume too much memory.