sklearn SGDClassifier kann nicht stoppen

Question

Ich verwende sklearn, um ein Modell zu trainieren. Der Zug dataset ist ungefähr 3000k, so verwende ich SGDClassifier. Das Feature ist nicht sehr gut, also weiß ich, dass es nicht konvergieren kann. Aber ich möchte SGDClassifier früh nach meiner Einstellung wie max_iter = 1000 stoppen. Soweit es mich betrifft, hat die Funktion SGDClassifier keinen Parameter wie max_iter. Wie kann ich es tun? Dies ist der Code.

Dies ist die Druckinformation.

Jede mögliche Hilfe geschätzt wird ...

Answer 1

Das ist seltsam, die standardmäßig in Scikit-Learn 0.18.2, n_iter wird auf 5 Epochen. Können Sie Ihre Frage bitte mit einem Skript aktualisieren, das es ermöglicht, das Verhalten mithilfe eines Spielzeug-Datasets (z. B. generiert mit numpy.random.randn oder ähnlichem) zu reproduzieren.

anzumerken, dass in Scikit-Learn-Master und 0,19 einmal freigesetzt, wird n_iter veraltet und durch max_iter und ein tol (beispielsweise bis 1e-3 eingestellt ist) ersetzt werden, um automatisch zu stoppen, wenn die Zielfunktion nicht mehr Fortschritte.

Answer 2

Die 20 Stunden laufen könnte nicht so seltsam sein, da Sie einen Datensatz von 3000k haben und Sie verwenden SGDClassifier, die langsam ist. Welchen Prozessor hast du?

Versuchen Sie es zu stoppen, indem Sie STRG + C verwenden, wenn Sie sich in Windows befinden. Verwenden Sie dann n_iter, um die Anzahl der gewünschten Iterationen zu steuern. Der Standardwert ist 5 jedoch.

Schließlich, wenn Sie ein Modell speichern möchten sehen hier:

Save and Load Machine Learning Models in Python with scikit-learn