Sebastian Pölsterl: scikit-sur... ノート

Sebastian Pölsterl: scikit-survival 0.28.0 リリース

Scikit-survival バージョン 0.28.0 がリリースされ、pandas DataFrame に加えて Polars DataFrame のサポートが導入されました。これは Narwhals DataFrame 抽象化レイヤーを通じて実現され、パフォーマンス機能が強化されています。このリリースでは、scikit-learn バージョン 1.9 との互換性も確保されています。scikit-survival 内のデータセットは、output_type 引数を指定することで Polars DataFrame として読み込むことができるようになりました。scikit-learn の set_output API により、トランスフォーマーは Polars DataFrame を返すことができ、scikit-survival の ClinicalKernelTransform と OneHotEncoder は現在、Polars と pandas の両方の入力を受け入れます。ClinicalKernelTransform は Polars の列タイプを認識し、順序付けのためにカテゴリ順序を指定できます。OneHotEncoder は文字列列を数値列に変換し、入力と同じタイプの DataFrame を返します。生存モデルは、Polars DataFrame を使用して適合および予測できるようになりましたが、これは eager DataFrame のみであり、lazy DataFrame は対象外です。このリリースでは、scikit-learn および Polars サポートに関する作業に対して 2 名の新しい貢献者が認められています。更新された依存関係には、Narwhals 2.0.1 および scikit-learn 1.9.0 が含まれます。Scikit-survival は pip または conda を介してインストールできます。