ML.NET加载来自枚举的数据返回空空IDataview

发布于 2025-01-23 10:45:49 字数 1212 浏览 3 评论 0原文

我正在尝试将数据库中的数据放入列表中,然后将其加载到IDATAVIEW中,但是每次Idataview都是空的。这是问题代码:

(IDataView training, IDataView test) LoadData(MLContext mlContext)
{
    var trainData = new List<ModelInput>();


    try
    {
        using var conn = new MySqlConnection(connection);

        conn.Open();

        var selectCmd = "SELECT user_id, movie_id, review_value FROM review";

        var command = new MySqlCommand(selectCmd, conn);

        var reader = command.ExecuteReader();

        while (reader.Read())
        {
            trainData.Add(new ModelInput
            {
                user_id = reader.GetInt32(0),
                movie_id = reader.GetInt32(1),
                review_value = reader.GetFloat(2),
            });
        }
    }
    catch (Exception ex)
    {
        Console.WriteLine(ex.Message);
    }

    var testDataPath = "recommendation-ratings-test.csv";

    IDataView trainingDataView = mlContext.Data.LoadFromEnumerable<ModelInput>(trainData);
    IDataView testDataView = mlContext.Data.LoadFromTextFile<ModelInput>(testDataPath, hasHeader: true, separatorChar: ',');

    return (trainingDataView, testDataView);
}

我找不到有关此问题以及如何解决问题的任何内容。请帮忙。

I am trying to put data from a database into a list and then loading it into an IDataView, but the IDataView is empty every time. Here is the problem code:

(IDataView training, IDataView test) LoadData(MLContext mlContext)
{
    var trainData = new List<ModelInput>();


    try
    {
        using var conn = new MySqlConnection(connection);

        conn.Open();

        var selectCmd = "SELECT user_id, movie_id, review_value FROM review";

        var command = new MySqlCommand(selectCmd, conn);

        var reader = command.ExecuteReader();

        while (reader.Read())
        {
            trainData.Add(new ModelInput
            {
                user_id = reader.GetInt32(0),
                movie_id = reader.GetInt32(1),
                review_value = reader.GetFloat(2),
            });
        }
    }
    catch (Exception ex)
    {
        Console.WriteLine(ex.Message);
    }

    var testDataPath = "recommendation-ratings-test.csv";

    IDataView trainingDataView = mlContext.Data.LoadFromEnumerable<ModelInput>(trainData);
    IDataView testDataView = mlContext.Data.LoadFromTextFile<ModelInput>(testDataPath, hasHeader: true, separatorChar: ',');

    return (trainingDataView, testDataView);
}

I cannot find anything only about this problem and about how to solve it. Please help.

如果你对这篇内容有疑问,欢迎到本站社区发帖提问 参与讨论,获取更多帮助,或者扫码二维码加入 Web 技术交流群。

扫码二维码加入Web技术交流群

发布评论

需要 登录 才能够评论, 你可以免费 注册 一个本站的账号。
列表为空,暂无数据
我们使用 Cookies 和其他技术来定制您的体验包括您的登录状态等。通过阅读我们的 隐私政策 了解更多相关信息。 单击 接受 或继续使用网站,即表示您同意使用 Cookies 和您的相关数据。
原文