Skip to content
Snippets Groups Projects
Commit 8345be11 authored by Hervé  MENAGER's avatar Hervé MENAGER
Browse files

use another strategy to avoid matplotlib backend error

Former-commit-id: d20b0efb39ae273a99bfdfa19d9fe216afe2ec7b
parent a66174d4
No related branches found
No related tags found
No related merge requests found
...@@ -6,7 +6,6 @@ import base64 ...@@ -6,7 +6,6 @@ import base64
from django.core.management import BaseCommand, CommandError from django.core.management import BaseCommand, CommandError
from django.forms.models import model_to_dict from django.forms.models import model_to_dict
import matplotlib.pyplot as plt import matplotlib.pyplot as plt
matplotlib.use('Agg')
import pandas as pd import pandas as pd
import numpy as np import numpy as np
from sklearn.decomposition import PCA from sklearn.decomposition import PCA
...@@ -55,6 +54,7 @@ class Command(BaseCommand): ...@@ -55,6 +54,7 @@ class Command(BaseCommand):
pca = PCA(n_components=2) pca = PCA(n_components=2)
principal_components = pca.fit_transform(x) principal_components = pca.fit_transform(x)
# compute correlation circle # compute correlation circle
plt.switch_backend('Agg')
variance_ratio = pd.Series(pca.explained_variance_ratio_) variance_ratio = pd.Series(pca.explained_variance_ratio_)
coef = np.transpose(pca.components_) coef = np.transpose(pca.components_)
cols = ['PC-'+str(x) for x in range(len(variance_ratio))] cols = ['PC-'+str(x) for x in range(len(variance_ratio))]
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment