Path: blob/master/Sequence Models/Week 1/Jazz improvisation with LSTM/__pycache__/music_utils.cpython-36.pyc
15918 views
3
��qZ� � @ s� d dl mZ d dlZd dljZd dlmZ d dl Z d dl
T d dl Z d dl
T d dlT d dlT ddd�Zdd d �Zd
d� Zdd� ZdS )� )�print_functionN)� RepeatVector)�*�< � c C s� |}t t| ��}tjjd� tj|||ftjd�}tj|||ftjd�}xzt|�D ]n}tjjt | �| �}| ||| � } xDt|�D ]8}
|| |
} |
dkr�d|||
| f< d|||
d | f< q�W qRW tj |dd�}|j
� }tj |�tj |�|fS )Nr )�dtype� ) �len�set�np�random�seed�zeros�bool�range�choice�swapaxes�tolist�asarray) �corpus�values_indices�mZTxZN_values�X�Y�iZ
random_idxZ corp_data�j�idx� r �D/home/jovyan/work/Week 1/Jazz improvisation with LSTM/music_utils.py�data_processing
s r �� � �?c C s~ |dk rzd} xl|j d�d dks2t|j d��dkrx| |kr`tjjdt|��}
||
j d�d }n|| |||�}| d7 } qW |S )a�
Helper function to fix the first value.
Arguments:
next_value -- predicted and sampled value, index between 0 and 77
x -- numpy-array, one-hot encoding of next_value
predict_and_sample -- predict function
indices_values -- a python dictionary mapping indices (0-77) into their corresponding unique value (ex: A,0.250,< m2,P-4 >)
abstract_grammars -- list of grammars, on element can be: 'S,0.250,<m2,P-4> C,0.250,<P4,m-2> A,0.250,<P4,m-2>'
duration -- scalar, index of the loop in the parent function
max_tries -- Maximum numbers of time trying to fix the value
Returns:
next_value -- process predicted value
g�h㈵��>r �,�R� � r )�splitr r r �randint) �modelZ
next_value�xZpredict_and_sampleZindices_values�abstract_grammars�duration� max_triesZ temperature�tries�randr r r �next_value_processing" s r/ c C sX t | �}tjd|t |�f�}x6t| �D ]*\}}||kr>t|� d|d||| f< q&W |S )z�
Convert a sequence (slice of the corpus) into a matrix (numpy) of one-hot vectors corresponding
to indices in values_indices
Arguments:
sequence -- python list
Returns:
x -- numpy-array of one-hot vectors
r g �?r )r r r � enumerate�print)�sequencer Z sequence_lenr) �t�valuer r r �sequence_to_matrixF s r5 c C s& t j| �} tj| d�} td�| �} | S )N�N r )�K�argmax�tf�one_hotr )r) r r r r: X s
r: )r r )r r! )�
__future__r Z
tensorflowr9 Z
keras.backend�backendr7 � keras.layersr �sys�music21�numpyr �grammar�
preprocess�qar r/ r5 r: r r r r �<module> s
$