using Microsoft.ML;
using Microsoft.ML.Data;
using System;
using System.Data.SqlClient;

namespace InQu.ML.Test.ONNX.FehlerAbschätzungLSTM
{
    class Program
    {
        static void Main(string[] args)
        {
            var mlContext = new MLContext();
            
            var x = new InputData[] { 
                        new InputData() { X = new float[,] { { 1, 2, 3, 4, 5, 6, 7 }, { 1, 2, 3, 4, 5, 6, 7 }, { 1, 2, 3, 4, 5, 6, 7 } } }                       
                    };

            IDataView dataView = mlContext.Data.LoadFromEnumerable<InputData>(x);

            var pipeline = mlContext.Transforms.ApplyOnnxModel(modelFile: @".\scikit-learn\AbschatzungFehlerLSTM.onnx", 
                            inputColumnNames: new[] { "input_layer",  }, 
                            outputColumnNames: new[] { "dense_3" });

            var model = pipeline.Fit(dataView);
            
            var predEngine = mlContext.Model.CreatePredictionEngine<InputData, OutputData>(model);

            OutputData prediction = new OutputData() { y = new float[0] };
            predEngine.Predict(x[0], ref prediction);
        }
    }

    public class InputData
    {
        [ColumnName("input_layer")]
        [VectorType(3,7)]
        public float[,] X { get; set; }        
    }

    public class OutputData
    {
        [ColumnName("dense_3")]
        [VectorType(1)]
        public float[] y { get; set; }
    }
}