Skip to content

Galaxy classification using Decision Tree and Random Forest Classifiers

Notifications You must be signed in to change notification settings

jayashree2699/Galaxy-Classification

Repository files navigation

Galaxy-Classification using Decision Tree Classifier and Random Forest Classifier

In this activity, we will be using machine learning to classify galaxies into three types based on their observed properties

  • ellipticals
  • spirals
  • galactic mergers

The features that we will be using to do our galaxy classification are:
  • colour index
  • adaptive moments
  • eccentricities
  • concentrations
  • Models Used

  • Decision Tree Classifier
  • Random Forest Classifier
  • Results and Discussions

    Confusion matrix of Decision Tree Classifier

    DscConfusionMatrix

    Confusion matrix of Random Forest Classifier

    RFConfusionMatrix

    It's clear from the below table that random forest outperforms decision tree classifier.

    comparison