Path: blob/main/C2 - Advanced Learning Algorithms/week4/optional labs/__pycache__/utils.cpython-310.pyc
3591 views
o
v�hd8 � @ s� d dl mZ d dlZd dlmZ d dlmZ d dl Z
d dl m Z m
Z
ej�d� dd� Zdd � Zd
d � Zd d
� Zdd� Zdd� Zdd� Zdd� Zdd� ZdS )� )�ImageN)�graphviz_layout)�Slider�Buttonz./deeplearning.mplstylec C sf d}t | �dkr
dS t| | dk �t | � }|dks |dkr dS | t�|� d| t�d| � S �Nr � )�len�sum�np�log2)�y�entropy� r �md:\TA\machine-learning-specialization-coursera\C2 - Advanced Learning Algorithms\week4\optional labs\utils.py�compute_entropy s &r c C s@ g }g }|D ]}| | | dkr|� |� q|� |� q||fS )Nr )�append)�X� node_indices�feature� left_indices�
right_indices�ir r r �
split_dataset s r c C s� t | ||�\}}| | || }}| | || }} | | || }
} d} t|�}
t| �}t| �}t|�t|� }t|
�t|� }|| || }|
| } | S )Nr )r r r )r r r r r r ZX_nodeZy_nodeZX_leftZy_leftZX_rightZy_rightZinformation_gainZ node_entropyZ left_entropyZ
right_entropyZw_leftZw_rightZweighted_entropyr r r �compute_information_gain% s r c C sB | j d }d}d}t|�D ]}t| |||�}||kr |}|}q
|S )Nr �����r )�shape�ranger )r r r Z num_features� best_featureZ
max_info_gainr Z info_gainr r r �get_best_split9 s
�r c C s� ||krd| d| }t |d| |� d S t| ||�}d| }t d||||f � t| ||�\} }
|�| |
|f� t| || d||d |� t| ||
d||d |� |S )N� �-z%s leaf node with indicesz%%s Depth %d, %s: Split on feature: %d�Leftr �Right)�printr r r �build_tree_recursive) r r r Z branch_name� max_depthZ
current_depth�tree�
formattingr r r r r r r$ I s r$ c
C s� dd� | D �}dd� |D �}t dd� |D �� \}}t|�}t|�}t�d||f�}d}|D ]} |�| |df� || jd 7 }q-|�t|t | � d �t|t | � d �f�}|S ) Nc S s g | ]}d | �qS )z
images/%d.pngr )�.0�idxr r r �
<listcomp>_ � z'generate_node_image.<locals>.<listcomp>c S s g | ]}t �|��qS r )r �open)r( �xr r r r* ` s c s s � | ]}|j V qd S �N)�size)r( r r r r � <genexpr>a s � z&generate_node_image.<locals>.<genexpr>�RGBr �
)
�zipr �maxr �new�paster/ �resize�intr )
r Z image_paths�images�widths�heights� total_width�
max_height�new_im�x_offset�imr r r �generate_node_image^ s .rA c C sp t �� }| ||g}t|�D ]
\}}|j|t|�d� q
|�dd� |�dd� t|dd�}t�� } t� d�}
|
�
d� t j |||
d d
d � |
j j
} | j�� j
} g d �| }
d
|
d|
d|
g}t|�D ]H\}}| || �\}}| ||f�\}}t|| �d }|d }t�|| || ||g�}|�
d� |�|j| d � |�d� |�|| � qd|
�d� t�� d S )N��imager r � �dot��prog�o �equalT�( ��ax�arrows� arrowsize�z Ear Shapez
Face ShapeZWhiskerszSplitting on %sz Left: %s = 1z
Right: %s = 0� � @rC �off)�nx�DiGraph� enumerate�add_noderA �add_edger �plt�figure�subplot�
set_aspect�draw_networkx_edges� transData� transform� transFigure�invertedr �axes�imshow�nodes�axis� set_title�show)r r r r �G� indices_listr) �indices�pos�figrL �trans�trans2� feature_name�ax_name�n�xx�yy�xa�ya�piesize�p2�ar r r �generate_split_vizr s6
rx c C s� t �� }|jdt| �d� d}d}t| �g}g d�}ddg}g } g }
t|�D ]I\} } | d d� }
|
D ]-}|j|t|�d� |�||� |�t|�� |d7 }| dkr_|
�d|t|| � � q2| �d || d � |d7 }q&| |
}t |d
d �}t
j d d
�}t
� d�}|�
d� t j|||ddd� |jj}|j�� j}t|�D ]Q\}}||| �\}}|||f�\}}|| d }|d }t
�|| || ||g�} | �
d� | �|j| d � | �d� z
| j|| dddd� W q� Y q�|�d� t
�� d S )Nr rB r rO zNon CatZCatrD z
Leaf node: %sz Split on: %srE rF )� r2 )�figsizerH rI TrJ rK � rQ rC rR g��������
�left)r �fontsize�loc)rS rT rV rA r rU rW r r4 r rX rY rZ r[ r\ r] r^ r_ r` ra rb rc rd re rf ) Z root_indicesr r& rg r) �rootZ
num_imagesrn Zy_nameZdecision_namesZ
leaf_namesr �levelrh ri �
node_namesrj rk rL rl rm rp rq rr rs rt ru rv rw r r r �generate_tree_viz� sX
�
r� c s� dd� � t �ddd�} � fdd�| D �}t�� \}}tjddd � |�d
� |�d � |�d � t�g d
��}|� | |�}|j
dddddd��t |d ddddd�}� �fdd�}|� |� |S )Nc S s: | dks| dkr
dS | t �| � d| t �d| � S r )r
r )�pr r r r
� s &z plot_entropy.<locals>.entropyr r �� c s g | ]}� |��qS r r )r( r� )r
r r r* � r+ z plot_entropy.<locals>.<listcomp>� �?)r} �bottomzp x H(p)r� zH(p))r� g�������?g�������?g� ��Q��?�red�d �F )�color�zorder�sg�������?)�valinit�valstepc s | }� |�}�� ||f� d S r. )� set_offsets)�valr- r �r
�scatterr r �update� s z plot_entropy.<locals>.update)
r
�linspacerX �subplots�subplots_adjustre �
set_xlabel�
set_ylabelra �plotr� r �
on_changed)Zp_arrayZh_arrayrk rL ZaxfreqZh_plot�sliderr� r r� r � plot_entropy� s
r� )�PILr ZnetworkxrS �matplotlib.pyplot�pyplotrX Znetworkx.drawing.nx_pydotr �numpyr
Zmatplotlib.widgetsr r �style�user r r r r$ rA rx r� r� r r r r �<module> s
% <