|
@@ -14,7 +14,7 @@ public class VovH24WeightModel extends Model {
|
|
|
private static final Logger LOGGER = LoggerFactory.getLogger(VovH24WeightModel.class);
|
|
|
|
|
|
|
|
|
- private Map<String, Map<String, Float>> model = new HashMap<>();
|
|
|
+ private Map<String, Map<String, Double>> model = new HashMap<>();
|
|
|
|
|
|
@Override
|
|
|
public int getModelSize() {
|
|
@@ -23,23 +23,24 @@ public class VovH24WeightModel extends Model {
|
|
|
|
|
|
@Override
|
|
|
public boolean loadFromStream(InputStreamReader in) throws Exception {
|
|
|
- Map<String, Map<String, Float>> initModel = new HashMap<>();
|
|
|
+ Map<String, Map<String, Double>> initModel = new HashMap<>();
|
|
|
try (BufferedReader reader = new BufferedReader(in)) {
|
|
|
StringBuilder buffer = new StringBuilder();
|
|
|
String line;
|
|
|
while ((line = reader.readLine()) != null) {
|
|
|
buffer.append(line).append("\n");
|
|
|
}
|
|
|
- initModel = JSONUtils.fromJson(buffer.toString(), new TypeToken<Map<String, Map<String, Float>>>() {
|
|
|
+ initModel = JSONUtils.fromJson(buffer.toString(), new TypeToken<Map<String, Map<String, Double>>>() {
|
|
|
}, initModel);
|
|
|
this.model = initModel;
|
|
|
} catch (Exception e) {
|
|
|
LOGGER.error("read vovh24 weight file error: ", e);
|
|
|
+ return false;
|
|
|
}
|
|
|
- return false;
|
|
|
+ return true;
|
|
|
}
|
|
|
|
|
|
- public Map<String, Float> getWeight(String key) {
|
|
|
+ public Map<String, Double> getWeight(String key) {
|
|
|
return model.get(key);
|
|
|
}
|
|
|
}
|