curiousily / todo-list-icon-classification-with-tensorflow-js

Use ReactJS and TensorFlow to suggest icons based on text in a ToDo list app

Hi there 👋

Hey, I am Venelin. Thanks for stopping by!

I am from an awesome little country called Bulgaria.

I work as a full-time Machine Learning engineer and write tutorials on basic and advanced topics (videos, posts, and code - lots of it).

You can:

• 📖 Read some of my tutorials
• 🍿 Subscribe to my YouTube channel
• ⏳ Prepare for the Machine Learning Interview with MLExpert
• ❤️ Sponsor me on GitHub

Have an awesome day!

todo-list-icon-classification-with-tensorflow-js's Issues

Attempted import error: 'formatLocale' is not exported from 'vega-functions'.

Downloaded project, ran npm i and got that error.

Attempted import error: 'formatLocale' is not exported from 'vega-functions'.

how to classify with more categories?

thanks for this tutorial! It's a great starting point.

I tried taking your example and training on more categories than just a binary classifier.
for this I added a block like this,
where you were returning a 2d array [1, 0]
I changed this to return 3 items like [1,0,0]

and then later in the prediction finding the right output.

when doing predictions with the same model later, I do get three values back from
model.predict.data()

but they don't map very well to the input data, I just get predictions for 1 or 2 columns, never the third etc.

Is there something else specific to your code that only works with a binary classifier?
like, I would need to add more layers to the model,
or use a different activation
or somehow be clearer about the inputShape that is passed in?