Choosing the right framework in Agentic AI - Cheat Sheet & Quick Revision

Recall & Review

beginner

What is a machine learning framework?

A machine learning framework is a set of tools and libraries that help you build, train, and deploy AI models more easily, like a toolbox for creating smart programs.

beginner

Name two popular machine learning frameworks.

Two popular frameworks are TensorFlow and PyTorch. They help you create AI models with ready-made tools and support.

beginner

Why is it important to choose the right framework?

Choosing the right framework saves time, makes coding easier, and fits your project needs better, like picking the right tool for a job.

intermediate

What factors should you consider when choosing a framework?

Consider ease of use, community support, compatibility with your data, speed, and the type of AI task you want to do.

intermediate

How does community support affect your choice of framework?

A strong community means more tutorials, help, and updates, making it easier to learn and solve problems.

Which of these is NOT a machine learning framework?

A React

B PyTorch

C TensorFlow

D Scikit-learn

React is a JavaScript library for building user interfaces, not a machine learning framework.

What is a key reason to pick a framework with a large community?

A It costs more money

B More bugs in the code

C It runs slower

D More tutorials and help available

A large community means more resources and support to help you learn and fix problems.

Which factor is LEAST important when choosing a framework?

A Ease of use

B Compatibility with your data

C Color of the framework's logo

D Speed of training models

The color of the logo does not affect how well the framework works for your project.

If you want to build a quick prototype, which framework feature is most helpful?

A Good documentation and examples

B Complex setup process

C Requires advanced coding skills

D Limited support

Good documentation and examples help you build prototypes faster and easier.

Which framework is known for being beginner-friendly and easy to learn?

A TensorFlow

B PyTorch

C Assembly language

D HTML

PyTorch is often praised for its simple and clear coding style, making it beginner-friendly.

Explain how you would choose the right machine learning framework for a new project.

Think about what matters most for your project and how the framework helps you.

Describe why community support is important when working with a machine learning framework.

Imagine learning something new without anyone to ask for help.

Practice

1. Which AI framework is best for beginners who want to learn with simple projects?

easy

A. Apache MXNet

B. PyTorch Lightning

C. Caffe

D. TensorFlow with Keras

Solution

1. Step 1: Identify beginner-friendly frameworks

   TensorFlow with Keras offers simple APIs and good tutorials for beginners.

2. Step 2: Compare with other options

   PyTorch Lightning and MXNet are more advanced; Caffe is less beginner-friendly.

3. Final Answer:

   TensorFlow with Keras -> Option D

4. Quick Check:

   Beginner-friendly = TensorFlow with Keras [OK]

Hint: Pick frameworks known for easy tutorials and simple APIs [OK]

Common Mistakes:

• Choosing complex frameworks for beginners
• Ignoring community support and tutorials
• Confusing advanced features with beginner ease

2. Which of the following is the correct way to import PyTorch in Python?

easy

A. import torch

B. import pytorch

C. from torch import pytorch

D. import Torch

Solution

1. Step 1: Recall PyTorch import syntax

   The official import statement is import torch.

2. Step 2: Check other options for errors

   'import Torch' uses incorrect capitalization; 'import pytorch' uses wrong module name; 'from torch import pytorch' tries to import a non-existent submodule.

3. Final Answer:

   import torch -> Option A

4. Quick Check:

   Standard import = import torch [OK]

Hint: Use official module name exactly as documented [OK]

Common Mistakes:

• Using wrong module names
• Trying to import submodules incorrectly
• Using uncommon aliases without reason

3. What will be the output of this code snippet using TensorFlow?

import tensorflow as tf
x = tf.constant([1, 2, 3])
y = tf.constant([4, 5, 6])
z = tf.add(x, y)
print(z.numpy())

medium

A. [1 2 3 4 5 6]

B. [4 10 18]

C. [5 7 9]

D. Error: TensorFlow add requires scalar inputs

Solution

1. Step 1: Understand tf.add operation

   tf.add adds element-wise values of two tensors of the same shape.

2. Step 2: Calculate element-wise addition

   [1+4, 2+5, 3+6] = [5, 7, 9]. The print statement outputs the numpy array.

3. Final Answer:

   [5 7 9] -> Option C

4. Quick Check:

   Element-wise add = [5 7 9] [OK]

Hint: Add tensors element-wise to get sum of each position [OK]

Common Mistakes:

• Confusing concatenation with addition
• Expecting scalar inputs only
• Misreading output format

4. You wrote this PyTorch code but get an error:

import torch
x = torch.tensor([1, 2, 3])
y = torch.tensor([4, 5])
z = x + y
print(z)

What is the main issue?

medium

A. Tensors have different shapes and cannot be added directly

B. You must convert tensors to numpy arrays before adding

C. PyTorch does not support tensor addition

D. You forgot to call .item() on tensors before adding

Solution

1. Step 1: Check tensor shapes

   x has shape (3,), y has shape (2,). They differ in length.

2. Step 2: Understand addition requirements

   PyTorch requires tensors to have compatible shapes for element-wise addition; these shapes are incompatible.

3. Final Answer:

   Tensors have different shapes and cannot be added directly -> Option A

4. Quick Check:

   Shape mismatch causes addition error [OK]

Hint: Check tensor shapes before adding [OK]

Common Mistakes:

• Assuming automatic broadcasting without matching shapes
• Converting unnecessarily to numpy
• Misusing .item() which extracts single values

5. You want to build an AI agent that can learn from text and images, and you want fast prototyping with easy debugging. Which framework should you choose?

hard

A. TensorFlow with low-level API

B. PyTorch with dynamic computation graphs

C. Scikit-learn

D. Theano

Solution

1. Step 1: Identify framework features needed

   Fast prototyping and easy debugging require dynamic computation graphs.

2. Step 2: Match features to frameworks

   PyTorch supports dynamic graphs and is popular for research and prototyping; TensorFlow low-level API is more complex; Scikit-learn is for classical ML, not deep learning; Theano is outdated.

3. Final Answer:

   PyTorch with dynamic computation graphs -> Option B

4. Quick Check:

   Dynamic graphs = PyTorch [OK]

Hint: Dynamic graphs help fast prototyping and debugging [OK]

Common Mistakes:

• Choosing static graph frameworks for prototyping
• Using classical ML libraries for deep learning tasks
• Picking outdated or unsupported frameworks