From 279d3e2eee1042c61a026bfa5f6d2cb2b0d09c90 Mon Sep 17 00:00:00 2001
From: Aleksandr Drozd <alexander.drozd@gmail.com>
Date: Mon, 21 Jan 2019 22:52:03 +0900
Subject: [PATCH] cleanup

---
 vecto/benchmarks/analogy/analogy.py | 12 ++++--------
 1 file changed, 4 insertions(+), 8 deletions(-)

diff --git a/vecto/benchmarks/analogy/analogy.py b/vecto/benchmarks/analogy/analogy.py
index 7e1a07c5..16dd4cae 100644
--- a/vecto/benchmarks/analogy/analogy.py
+++ b/vecto/benchmarks/analogy/analogy.py
@@ -1,7 +1,6 @@
 import datetime
 import os
 import uuid
-import numpy as np
 import logging
 import progressbar
 # from tqdm import tqdm
@@ -9,7 +8,10 @@
 from vecto.data import Dataset
 from ..base import Benchmark
 from .io import get_pairs
-from .solvers import *
+from .solvers import LinearOffset, LRCos, PairDistance
+from .solvers import ThreeCosAvg, ThreeCosMul, ThreeCosMul2
+from .solvers import SimilarToAny, SimilarToB
+
 
 logger = logging.getLogger(__name__)
 
@@ -64,7 +66,6 @@ def __init__(self,
 
         self.stats = {}
 
-
         # this are some hard-coded bits which will be implemented later
         self.result_miss = {
             "rank": -1,
@@ -102,9 +103,7 @@ def __init__(self,
     #         distances[i] = scores[ids_max[i + 1]]
     #     return distances.mean()
 
-
     def run_category(self, pairs):
-
         details = []
         kfold = sklearn.model_selection.KFold(n_splits=len(pairs) // self.size_cv_test)
         cnt_splits = kfold.get_n_splits(pairs)
@@ -231,6 +230,3 @@ def get_result(self, embeddings, path_dataset):  # , group_subcategory=False
             embeddings.normalize()
         results = self.run(embeddings, path_dataset)  #group_subcategory
         return results
-
-
-