使用pytorch将数据点分为两个类

我在（x，y）域上有一些数据点，它们应该被分成两个类，这里用x和O表示，如下所示

import numpy as np
import matplotlib.pyplot as plt
x1 = np.array([0.1,0.3,0.1,0.6,0.4,0.6,0.5,0.9,0.4,0.7])
x2 = np.array([0.1,0.4,0.5,0.9,0.2,0.3,0.6,0.2,0.4,0.6])
c=np.array([ 1,1,1,1,1,0,0,0,0,0 ])
plt.plot(x1[c==0], x2[c==0], 'bo')
plt.plot(x1[c==1], x2[c==1], 'rx')

现在我想找到一种方法，这样我就可以找到“最佳拟合曲线”来分离像这样的曲线

首先，我想我可能会尝试最近邻法，但我被告知它在这里不适用，有一种更简单的方法，用ANN和PyTorch来实现它，但我不明白怎么做。 关于使用什么和/或如何使用它以及我应该检查什么来评估我的准确性，有什么想法吗？？积分太少了，这里的想法是什么

---

提前谢谢大家

这似乎是一个基本情况，而神经网络可能是完成这项任务的一种过度手段。此外，如果您以后想像您发布的那样进行一些可视化，您可能应该采用更基本的方法

您可以尝试线性分类器（支持向量机、线性回归等）

检查下面的链接