SV12/ERA_Session13
0
1o
2 �"+fj � @ s� d dl Zd dlZd dlmZ d dlZd dlZd dlm Z d dl3m Z dd� Z dd� Z
d d d4�Zd d � Zd
d� Zdd� Zdd� Zdd� Zd dd�Zdd� Zd!d d �ZdS )"� N��summary)�LRFinderc C s8 t | |||d�}|j||||dd� |�� |�� d S )N)�model� optimizer� criterion�device�exp)Z train_loader�start_lr�end_lr�num_iterZ step_mode)r Z5range_test�plot�reset) r r r r Z trainloader�numiterZstartlrZendlr� lr_finder� r �=C:\Users\Shubham\Documents\ERA\ERA Session-13\utils\common.py�find_lr s �� r c C s( t jjj| |||d| ddddd� }|S )N� �d F�linear) r �max_lr�steps_per_epoch�epochs� pct_start�6div_factor� three_phase�final_div_factor�anneal_strategy)�torch�optim� lr_scheduler�7OneCycleLR)r �maxlr�stepsr � schedulerr r r � one_cycle_lr s � r& � c s\ t | j�D ]&\� }tj� fdd�t | j�D �|d�}ttj| j| dd�� t �8|� qd S )Nc s g | ]9\}}|� kr|�qS r r )�.0�n�x��cr r �10<listcomp>/ s z5show_random_images_for_each_class.<locals>.<listcomp>)�k)r � � � )�axes) � enumerate�classes�random�sample�targets�
show_img_grid�np� transpose�data�plt�title)�11train_dataZnum_images_per_class�clsZ rand_targetsr r+ r �!show_random_images_for_each_class, s � �r@ c C s\ z t j�| �� �� �}W n t�| �} t j�| �}Y tjdd� t� |�12ddd�� d S )N��13 rB ��figsizer0 r1 r ) � torchvision�utils� make_grid�cpu�detachr �14from_numpyr< �figure�imshow�permute)r; Zgrid_imgr r r r8 6 s 15 r8 c C s t t| ��\}}t|� d S )N)�next�iterr8 )Z data_loaderr; �targetr r r �show_random_imagesA s rQ c C s t | |dddfg d�dd� d S )Nr/ � )�16input_size� output_size�17num_params� kernel_sizer0 )r rS Z col_names�verboser )r �18batch_sizer r r �show_model_summaryF s 1920�rY c C s� t �| d | d � t �| d | d � t �ddg� t �d� t �d� t �d� t �� t �| d | d � t �| d | d21 � t �d d g� t �d� t �d
� t �d� t �� d S )N�epochZ trainlossZtestlossz22Train LosszValidation Loss�Epochs�LosszLoss vs EpochsZtrainaccZtestaccz Train AcczValidation Acc�AccuracyzAccuracy vs Epochs)r< r
�legend�xlabel�ylabelr= �show)�resultsr r r �
lossacc_plotsO s 232425262728 rc c C s> t �t|�| d � t �d� t �d� t �d� t �� d S )N�lrr[ z
Learning RatezLearning Rate vs Epochs)r< r
�ranger_ r` r= ra )rb �lengthr r r �lr_plotsa s29 303132 rg rB c
C s� g g g }}}t �� �T |D ]I\}}|�|�|�|�}}| |�} | jddd�}33t �|34�� |k��� } | D ]!} t|�|kr@ n|�|| � |�|| � |�|35| �� � q6qW d � n1 scw Y |||fS )Nr0 T��dim�keepdim)r �no_grad�to�argmax�argwhere�squeeze�len�append)
r Z36testloaderr Z mis_countZmisimgsZmistgtsZmispredsr; rP �output�predZ
misclassified�idxr r r �get_misclassifiedi s" 37 �� �38 ru c C s� | � � g }t�� �H |D ]=\}}|�|�|�|�}}t||�D ] \}}|�d�}| |�} | jddd�}39|40|krA|�|||41f� q!t|�|krJ nq
W d � n1 sUw Y |d|� S )z�42 Function to run the model on test set and return misclassified images43 :param model: Network Architecture44 :param device: CPU/GPU45 :param test_loader: DataLoader for test set46 r r0 Trh N) �evalr rk rl �zip� unsqueezerm rq rp ) r r Z test_loader�countZmisclassified_datar; rP �image�labelrr rs r r r �get_misclassified_data� s$ 47 48� �� � r| rA r1 r c 49 C s� t j|d�}t| �}t|�D ]L}t �|||d � | | d �� �d�} |d ur,|| �} t �t� | d�� t �50d|| | d � � � d|| | d � � � �� t � g � t �
g � qd S ) NrC r0 r rH )r0 r1 r zLabel: z 51 Prediction: r1 )r< rK rp re �subplotro rl rL r9 r: r= �item�xticks�yticks)52r; r4 �size�rows�colsZ
inv_normalize�figZnumber_of_samples�i�imgr r r �plot_misclassified� s 653 �r� )r' )rB )rA r1 r N)�numpyr9 r5 �matplotlib.pyplot�pyplotr< r rE Z torchinfor Ztorch_lr_finderr r r& r@ r8 rQ rY rc rg ru r| r� r r r r �<module> s$ 5455 56)