How it works...

The data is comprised of 1,599 red wines and 4,898 white wines with 11 features and an output label that can be used during training.

The following is a list of features/attributes:

  • Fixed acidity
  • Volatile acidity
  • Citric acid
  • Residual sugar
  • Chlorides
  • Free sulfur dioxide
  • Total sulfur dioxide
  • Density
  • pH
  • Sulphates
  • Alcohol

The following is the output label:

  • quality (a numeric value between 0 to 10)

Get Apache Spark 2.x Machine Learning Cookbook now with the O’Reilly learning platform.

O’Reilly members experience books, live events, courses curated by job role, and more from O’Reilly and nearly 200 top publishers.