线性回归速度增加

Question

我是机器学习的新手，这是我第一次在数据集上创建线性回归模型（这对我来说是很重要的一步）。我已经创建了参考行并对其进行了调整。唯一的问题是它太慢了。有没有我可以使用的任何代码或更好的方法。如果您有机会修改我的代码，那就太好了。 谢谢。

#Build the JAR with maven 
FROM maven:3.6.3-jdk-11 AS build  

COPY . /app
RUN mvn -f /app/pom.xml clean package #/root/.m2 should be the default location


#Second stage, in order to minimize the image footprint
FROM gcr.io/distroless/java:11  
COPY --from=build /app/service1/target/service1.jar /app/service1.jar 
ENTRYPOINT ["java","-jar","/app/service1.jar"]

输出为= 10/10，损失：nan

我的预期输出应该每次都会降低我的损失函数值。

我要迭代此回归至少1000次。

我有1660Ti，i7第9代，16GB ram笔记本电脑