[ml-completion] Update java model: 0.1.1

GitOrigin-RevId: e1b80c2f6719a1c4437aba41ebc97ea568fbf4f6
This commit is contained in:
Vitaliy.Bibaev
2019-12-28 15:10:32 +03:00
committed by intellij-monorepo-bot
parent 43e64e1c56
commit b92c0da149
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.10" />
<properties include-transitive-deps="false" maven-id="org.jetbrains.intellij.deps.completion:completion-ranking-java:0.1.1" />
<CLASSES>
<root url="jar://$MAVEN_REPOSITORY$/org/jetbrains/intellij/deps/completion/completion-ranking-java/0.0.10/completion-ranking-java-0.0.10.jar!/" />
<root url="jar://$MAVEN_REPOSITORY$/org/jetbrains/intellij/deps/completion/completion-ranking-java/0.1.1/completion-ranking-java-0.1.1.jar!/" />
</CLASSES>
<JAVADOC />
<SOURCES>
<root url="jar://$MAVEN_REPOSITORY$/org/jetbrains/intellij/deps/completion/completion-ranking-java/0.0.10/completion-ranking-java-0.0.10-sources.jar!/" />
<root url="jar://$MAVEN_REPOSITORY$/org/jetbrains/intellij/deps/completion/completion-ranking-java/0.1.1/completion-ranking-java-0.1.1-sources.jar!/" />
</SOURCES>
</library>
</orderEntry>

View File

@@ -1,16 +1,16 @@
// 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.MLCompletionModel
import com.intellij.internal.ml.*
import com.intellij.internal.ml.completion.CompletionRankingModelBase
import com.intellij.internal.ml.completion.JarCompletionModelProvider
import com.intellij.lang.Language
import com.jetbrains.completion.ranker.model.java.MLGlassBox
class JavaMLRankingProvider : JarCompletionModelProvider("Java", "java_features") {
override fun createModel(metadata: ModelMetadata): DecisionFunction {
return object : CompletionRankingModelBase(metadata) {
override fun predict(features: DoubleArray?): Double = MLCompletionModel.makePredict(features)
override fun predict(features: DoubleArray?): Double = MLGlassBox.makePredict(features)
}
}