# ---
# jupyter:
#   jupytext:
#     text_representation:
#       extension: .py
#       format_name: light
#       format_version: '1.5'
#       jupytext_version: 1.14.4
#   kernelspec:
#     display_name: Python 3
#     language: python
#     name: python3
# ---

# +
import numpy as np
from numpy . linalg import svd
from itertools import product
from TRG import Z_TRG

A = []
B = np.linspace(0,1.6,56)
for i in B:
    A.append(Z_TRG(i, 6 , 3 ))
# -

import matplotlib.pyplot as plt
plt.figure(figsize=[8,5])
plt.plot(B,np.log(A),'-o',label='Tensor renormalization group');
plt.xlabel('Coupling constant',fontsize=18)
plt.ylabel('Free energy per unit site',fontsize=18)
plt.legend(fontsize=16);
plt.savefig('TRG.png')

StarcoderdataJupyterScriptsDedupFiltered