Ich werde es denen überlassen, die am selben Ort festsitzen. Bei der Verwendung von PyTorch wurde der folgende Fehler angezeigt.
RuntimeError: Expected object of scalar type Float but got scalar type Double for argument #4 'mat1'
Als Ergebnis verschiedener Untersuchungen scheint das Problem darin zu bestehen, dass bei der Konvertierung in den Tensortyp die Zahl im Tensor zum Typ fackel.double wird. (Es gibt viele Methoden in der PyTorch-Klasse, die auf dem Typ ** torch.float ** basieren.)
Damit
X_train = torch.from_numpy(X_train)
y_train = torch.from_numpy(y_train)
X_test = torch.from_numpy(X_test)
y_test = torch.from_numpy(y_test)
X_train = torch.from_numpy(X_train).float()
y_train = torch.from_numpy(y_train).long()
X_test = torch.from_numpy(X_test).float()
y_test = torch.from_numpy(y_test).long()
Es scheint, dass Sie es mit .float () oder .long () wie folgt konvertieren sollten. (.long () ist eine Konvertierung in ein Label)
Verweise: 2. PyTorch Tensol & Datentyp Cheet Sheet