Merge pull request #71 from myslak71/develop

Develop
myslak71 · Feb 23, 2019 · c16e9e6 · c16e9e6
2 parents 3493c46 + aaed9ab
commit c16e9e6
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 11 deletions.
diff --git a/common/utils/count_percentage.py b/common/utils/count_percentage.py
@@ -42,4 +42,5 @@ def count_percentage(duel, algorithm):
         algorithm.parameters.update({'random_state': 0})
     model = algorithms[algorithm.get_name_display()](**algorithm.parameters)
     model.fit(x_train, y_train)
+
     return float(Decimal(100 * model.score(x_test, y_test)).quantize(Decimal('.001')))
diff --git a/duels/api/serializers.py b/duels/api/serializers.py
@@ -43,15 +43,8 @@ class Meta:
         fields = ('__all__')
 
     def to_representation(self, instance):
-        algorithm = super().to_representation(instance)
+        algorithm_repr = super().to_representation(instance)
         if self.context.get('include_algorithm_name'):
-            choice = dict(Algorithm._meta.get_field('name').choices)[algorithm['name']]
-            algorithm['name_display'] = choice
-        return algorithm
-
-
-class DefaultAlgorithmSerializer(serializers.ModelSerializer):
-    class Meta:
-        model = DefaultAlgorithm
-        fields = ('__all__')
-        read_only_fields=('__all__')
+            algorithm_name = dict(Algorithm._meta.get_field('name').choices)[algorithm_repr['name']]
+            algorithm_repr['name_display'] = algorithm_name
+        return algorithm_repr