Classifying Hydrocarbons

I developed a machine learning Python program using the sklearn module that classifies a simple single-chain hydrocarbon  into categories based on the type of bonds it has given the number of carbons and hydrogens (alkanes have single bonds, alkenes have double bonds and alkynes have triple bonds).

Although there are easy ways to tell what type of bonds a simple single-chain hydrocarbon has given the number of carbons and hydrogens, I wanted to use a training set and a Linear SVM(Linear Support Vector Machine) to classify them in order to introduce myself to machine learning.

Link to download Python code

Here is a picture of the output of my program: