1. Learn
    2. /
  2. Courses
    3. /
  3. Introduction to Deep Learning with PyTorch
Connected

Exercise

From regression to multi-class classification

The models you have seen for binary classification, multi-class classification and regression have all been similar, barring a few tweaks to the model.

Start building a model for regression, and then tweak the model to perform a multi-class classification.

Instructions 1/2

undefined XP
  • 1
    • Create a 4-layer linear network that takes 11 input features from input_tensor and produces a single regression output.
  • 2
    • Update the network provided to perform a multi-class classification with four outputs.