Usage:

Step 1: Generate data with 

python heat.py


Step 2: Reconstruct heat equation using a CNN with:

python cnn_heat.py