train_unsupervised.html 1.3 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344
  1. <!DOCTYPE html>
  2. <html>
  3. <head>
  4. <meta charset="UTF-8">
  5. <meta name="viewport" content="width=device-width, initial-scale=1, minimum-scale=1.0, maximum-scale=1.0, user-scalable=no">
  6. </head>
  7. <body>
  8. <script type="module">
  9. const printVector = function(predictions, limit) {
  10. limit = limit || Infinity;
  11. for (let i=0; i<predictions.size() && i<limit; i++){
  12. let prediction = predictions.get(i);
  13. console.log(predictions.get(i));
  14. }
  15. }
  16. const trainCallback = (progress, loss, wst, lr, eta) => {
  17. console.log([progress, loss, wst, lr, eta]);
  18. };
  19. import {FastText, addOnPostRun} from "./fasttext.js";
  20. addOnPostRun(() => {
  21. let ft = new FastText();
  22. ft.trainUnsupervised("fil9", 'skipgram', {
  23. 'lr':0.1,
  24. 'epoch':1,
  25. 'loss':'ns',
  26. 'wordNgrams':2,
  27. 'dim':50,
  28. 'bucket':200000
  29. }, trainCallback).then(model => {
  30. let wordsInformation = model.getWords();
  31. printVector(wordsInformation[0], 30); // words
  32. printVector(wordsInformation[1], 30); // frequencies
  33. });
  34. });
  35. </script>
  36. </body>
  37. </html>