自动模型选择

时间:2018-02-27 12:08:25

标签: python-3.x machine-learning classification regression

我正在从头开始编写一个机器学习主算法,用户只需输入训练和测试数据,我想知道有没有办法自动决定使用什么算法:回归与分类

例如, (假设最后一列始终是输出,它始终是一个数字)

如果我们可以搜索最后一列并通过查看它们是离散类标签还是连续值来确定它是什么模型。 怎么会这样呢?

如果不是这种方法,还有更好的方法吗?

它是在python3中。

谢谢。

2 个答案:

答案 0 :(得分:1)

type of target using sk-learn

我们可以使用typeoftarget()函数来获取targget的类型:continuous或label,因此可以根据它来确定它是什么类型的问题。

@Vivekkumar提供的答案正如我所需要的那样。 谢谢!

可能并不总是有效。我们可以愚弄它仍然需要更好。但是对于最好的情况,我们可以为此运行ML算法,并将其用作模型以更好地训练它。

答案 1 :(得分:0)

以下代码应该有助于确定最后一列中的值是否可以转换为浮点数:

s_1 = 'text'
s_2 = '1.23'

for s in [s_1, s_2]:
    try:
        f = float(s)
        print(s, 'conversion to float was OK, value:', f)
    except:
        print(s, 'could not be converted to a number')