diff --git a/ippisite/ippidb/management/commands/pca.py b/ippisite/ippidb/management/commands/pca.py index 20fc2c4a002658554ac1bcec3678eccc33484c6d..0a626d1b55ef2d6c7354f2ee1a747485bdf4fffe 100644 --- a/ippisite/ippidb/management/commands/pca.py +++ b/ippisite/ippidb/management/commands/pca.py @@ -6,7 +6,6 @@ import base64 from django.core.management import BaseCommand, CommandError from django.forms.models import model_to_dict import matplotlib.pyplot as plt -matplotlib.use('Agg') import pandas as pd import numpy as np from sklearn.decomposition import PCA @@ -55,6 +54,7 @@ class Command(BaseCommand): pca = PCA(n_components=2) principal_components = pca.fit_transform(x) # compute correlation circle + plt.switch_backend('Agg') variance_ratio = pd.Series(pca.explained_variance_ratio_) coef = np.transpose(pca.components_) cols = ['PC-'+str(x) for x in range(len(variance_ratio))]