DEV Community

Cover image for Machine learning in Scratch?? πŸ±πŸ’‘
Pascal Thormeier
Pascal Thormeier

Posted on • Updated on

Machine learning in Scratch?? πŸ±πŸ’‘

Wait, really?

Yes, absolutely! Scratch has all the necessary tools for a working linear regression with gradient descent. With a few nifty tricks, we can even visualize how the algorithm works!

What is Scratch, though?

Scratch is a project of the Scratch Foundation, in collaboration with the Lifelong Kindergarten Group at the MIT Media Lab. It is available for free at Scratch is a tool to teach programming by offering a GUI where instructions are combined via drag&drop. Sprites and backgrounds allow for a user interface with animations and user input. One can create all kinds of programs, like simple interactive movies, entire games, or, well, machine learning algorithms.

Some example code in Scratch

Machine learning in Scratch? Why would you even do this?

Because it's possible. I wanted to explore the boundaries of Scratch and figure out new ways to solve certain problems. It made me understand the algorithm better and think outside the box. Besides: I've got a little history in implementing those things in languages or tools you wouldn't expect:

This post also explains how linear regression works and how to implement it.

Ok, let's implement linear regression in Scratch then!

Awesome! First, I need some data. For linear regression, that's a bunch of XY-coordinates. Since Scratch doesn't have nested lists, I keep them in separate lists. A third list keeps track of the error values, so we can inspect or display it later.

The lists in Scratch's side bar

I'm generating the data now, because why not. I could simply generate random data, but some kind of correlation would be nice. UVM has a good formula for that. First, xx and yy are set randomly. Then, a new yy is calculated with this formula, with rr as the "rate of correlation":

y=xβˆ—r+yβˆ—1βˆ’r2 y = x*r+y*\sqrt{\smash[b]{1-r^2}}

I create a new sprite with a single red dot in the center and tell it to calculate it's own coordinates as soon as it's created. I can then clone this point sprite to create new as many data points as I want:

The code for the data point sprite

The nice thing about this: The data is already visualized. This comes more or less out of the box.

I now introduce the variable m and set it to 100, to know how many data points I need. Next, I create a "main" sprite that's empty, basically the main entry point of my program. Think of it like Java's public static void main(String[] args). There I reset all the variables, set the parameters I need for the linear regression/gradient descent and clone the data point m times:

The main code to kick off the algorithm

And that's the result so far:

Data points, neatly distributed with some correlation

Next, I create a sprite called "error point" to plot the error rate. The "error point" is similar to the data point sprite. Also, since the "errors" list is already public, I introduce two more variables: maxIter for the total number of iterations of gradient descent and currentIter to know which iteration I'm currently at. For each iteration, I create a clone of the "error point" and tell it to adjust itself, to get a nice plot of the error rate:

Error point code

Next, I introduce a sprite called "line". This is where the main logic will happen. The line can basically adjust itself, rotate and move, whatever its internals tell it to do.

Linear regression basically tells me the coefficients c0c_0 and c1c_1 of a linear function. I can use those to calculate a single point (i.e. (x,c0+c1βˆ—x)(x, c_0 + c_1 * x) ). To calculate the angle Ξ±\alpha of the hypotenuse, I can use this formula:

tan⁑α=sideadjacentside \tan \alpha = \frac{side}{adjacent side}


α=arctan⁑sideadjacentside \alpha = \arctan \frac{side}{adjacent side}

When I assume xx as 1 and neglect c0c_0 (this is only moving the line, not rotating it), I can calculate the angle of the line as this:

α=arctan⁑c0 \alpha = \arctan c_0

This is then part of the iteration of gradient descent in the "line" sprite:

Line sprite, basic reset code, and the loop for gradient descent

So far so good. I've got data points, I've got an error plot, I've got a line that I can move around. Now for the fun part: The linear regression with gradient descent itself.

I've worked with two lists for x's and y's since Scratch doesn't allow for nested lists. Minor inconvenience, but nothing that'll stop me.

So, first, I'll give the variables used in the loop some default values:

Default values in gradient descent loop

Next up: Looping through the data points to calculate the different between predicted Y and actual Y:

Looping through all data points

Next, I use the sums descentSumC0 and descentSumC1 to calculate new versions of c0 and c1 and set them simultaneously:

Setting of C0 and C1

After this block ran through, the line is adjusted and a new calculation is done. Until the maximum number of iterations is reached.

And this works?


Animation of Scratch executing linear regression with gradient descent

For the full code and to try it yourself, here's the Scratch project:

I hope you enjoyed reading this article as much as I enjoyed writing it! If so, leave a ❀️ or a πŸ¦„! I write tech articles in my free time and like to drink coffee every once in a while.

If you want to support my efforts, buy me a coffee β˜• or follow me on Twitter 🐦! You can also support me directly via Paypal!

Buy me a coffee button

Discussion (1)

milg15 profile image
Maria Isabel Lopez

So cool!! Scratch it's a very lovely tool and I love seeing some more advance uses for it !