Regression Model for Superconductor Critical Temperature Using Python and TensorFlow Take 1

Template Credit: Adapted from a template made available by Dr. Jason Brownlee of Machine Learning Mastery.

SUMMARY: The purpose of this project is to construct a predictive model using various machine learning algorithms and to document the end-to-end steps using a template. The Superconductor Critical Temperature dataset is a regression situation where we are trying to predict the value of a continuous variable.

INTRODUCTION: The research team wishes to create a statistical model for predicting the superconducting critical temperature based on the features extracted from the superconductor’s chemical formula. The model seeks to examine the features that can contribute the most to the model’s predictive accuracy.

In this Take1 iteration, we will construct and tune machine learning models for this dataset using TensorFlow with five layers. We will observe the best result that we can obtain using the tuned models with the validation and test datasets.

ANALYSIS: The baseline performance of the TensorFlow algorithm achieved an RMSE benchmark of 11.109. After a series of tuning trials, the TensorFlow model processed the validation dataset with an RMSE score of 10.564. When we applied the TensorFlow model to the previously unseen test dataset, we obtained an RMSE score of 10.540.

CONCLUSION: In this iteration, the TensorFlow model appeared to be a suitable algorithm for modeling this dataset. We should consider using the algorithm for further modeling.

Dataset Used: Superconductivity Data Set

Dataset ML Model: Regression with numerical attributes

Dataset Reference:

One potential source of performance benchmarks:

The HTML formatted report can be found here on GitHub.