embedding_android_6.kt

@Throws(IOException::class)
private fun loadModelFile(): MappedByteBuffer {
    val MODEL_ASSETS_PATH = "model.tflite"
    val assetFileDescriptor = assets.openFd(MODEL_ASSETS_PATH)
    val fileInputStream = FileInputStream(assetFileDescriptor.fileDescriptor)
    val fileChannel = fileInputStream.channel
    val startoffset = assetFileDescriptor.startOffset
    val declaredLength = assetFileDescriptor.declaredLength
    return fileChannel.map(FileChannel.MapMode.READ_ONLY, startoffset, declaredLength)
}

fun classifySequence ( sequence : Array<DoubleArray> ): FloatArray {
    val interpreter = Interpreter( loadModelFile() )
    val inputs : Array<Array<FloatArray>> =  arrayOf(
        sequence.map{
            it.map {
                it.toFloat()
            }.toFloatArray()
        }.toTypedArray()
    )
    val outputs : Array<FloatArray> = arrayOf( floatArrayOf( 0.0f , 0.0f ) )
    interpreter.run( inputs , outputs )
    return outputs[0]
}