|
1 | 1 | package com.medallia.word2vec; |
2 | 2 |
|
3 | | -import java.io.DataInput; |
4 | | -import java.io.DataInputStream; |
5 | 3 | import java.io.File; |
6 | 4 | import java.io.FileInputStream; |
7 | 5 | import java.io.IOException; |
8 | 6 | import java.nio.ByteOrder; |
9 | 7 | import java.nio.FloatBuffer; |
10 | 8 | import java.nio.MappedByteBuffer; |
11 | 9 | import java.nio.channels.FileChannel; |
12 | | -import java.nio.file.Files; |
13 | 10 | import java.util.ArrayList; |
14 | 11 | import java.util.List; |
15 | 12 |
|
16 | | -import org.apache.commons.io.input.SwappedDataInputStream; |
17 | | - |
18 | 13 | import com.google.common.annotations.VisibleForTesting; |
19 | 14 | import com.google.common.base.Preconditions; |
20 | 15 | import com.google.common.collect.ImmutableList; |
21 | 16 | import com.google.common.collect.Lists; |
22 | 17 | import com.google.common.primitives.Doubles; |
23 | 18 | import com.medallia.word2vec.thrift.Word2VecModelThrift; |
24 | 19 | import com.medallia.word2vec.util.Common; |
25 | | -import org.slf4j.Logger; |
26 | | -import org.slf4j.LoggerFactory; |
| 20 | +import org.apache.log4j.Logger; |
27 | 21 |
|
28 | 22 | /** |
29 | 23 | * Represents the Word2Vec model, containing vectors for each word |
|
37 | 31 | * @see {@link #forSearch()} |
38 | 32 | */ |
39 | 33 | public class Word2VecModel { |
40 | | - static Logger logger = LoggerFactory.getLogger(Word2VecModel.class); |
| 34 | + static Logger logger = Logger.getLogger(Word2VecModel.class); |
41 | 35 |
|
42 | 36 | final List<String> vocab; |
43 | 37 | final int layerSize; |
|
0 commit comments