[ml-completion] Update Java model

GitOrigin-RevId: a327a3d58172de15ec290274201a64b6e78f2d56
This commit is contained in:
Vitaliy.Bibaev
2019-12-02 12:57:38 +03:00
committed by intellij-monorepo-bot
parent abc2c19258
commit a1614b3696
2 changed files with 5 additions and 5 deletions

View File

@@ -57,13 +57,13 @@
<orderEntry type="module" module-name="intellij.platform.diff.impl" />
<orderEntry type="module-library">
<library name="completion-ranking-java" type="repository">
<properties include-transitive-deps="false" maven-id="org.jetbrains.intellij.deps.completion:completion-ranking-java:0.0.7" />
<properties include-transitive-deps="false" maven-id="org.jetbrains.intellij.deps.completion:completion-ranking-java:0.0.10" />
<CLASSES>
<root url="jar://$MAVEN_REPOSITORY$/org/jetbrains/intellij/deps/completion/completion-ranking-java/0.0.7/completion-ranking-java-0.0.7.jar!/" />
<root url="jar://$MAVEN_REPOSITORY$/org/jetbrains/intellij/deps/completion/completion-ranking-java/0.0.10/completion-ranking-java-0.0.10.jar!/" />
</CLASSES>
<JAVADOC />
<SOURCES>
<root url="jar://$MAVEN_REPOSITORY$/org/jetbrains/intellij/deps/completion/completion-ranking-java/0.0.7/completion-ranking-java-0.0.7-sources.jar!/" />
<root url="jar://$MAVEN_REPOSITORY$/org/jetbrains/intellij/deps/completion/completion-ranking-java/0.0.10/completion-ranking-java-0.0.10-sources.jar!/" />
</SOURCES>
</library>
</orderEntry>

View File

@@ -1,7 +1,7 @@
// Copyright 2000-2019 JetBrains s.r.o. Use of this source code is governed by the Apache 2.0 license that can be found in the LICENSE file.
package com.intellij.codeInsight.completion.ml
import com.completion.ranker.model.java.MLWhiteBox
import com.completion.ranker.model.java.MLCompletionModel
import com.intellij.internal.ml.*
import com.intellij.internal.ml.completion.CompletionRankingModelBase
import com.intellij.internal.ml.completion.JarCompletionModelProvider
@@ -10,7 +10,7 @@ import com.intellij.lang.Language
class JavaMLRankingProvider : JarCompletionModelProvider("Java", "java_features") {
override fun createModel(metadata: ModelMetadata): DecisionFunction {
return object : CompletionRankingModelBase(metadata) {
override fun predict(features: DoubleArray?): Double = MLWhiteBox.makePredict(features)
override fun predict(features: DoubleArray?): Double = MLCompletionModel.makePredict(features)
}
}