Add citation (#383)

CITATION.cff

@@ -5,8 +5,6 @@ title: "PyTorch Frame: A Deep Learning Framework for Tabular Data"
 authors:
 - family-names: "Hu"
   given-names: "Weihua"
-- family-names: "Fey"
-  given-names: "Matthias"
 - family-names: "Yuan"
   given-names: "Yiwen"
 - family-names: "Zhang"
@@ -17,6 +15,10 @@ authors:
   given-names: "Kaidi"
 - family-names: "Kocijan"
   given-names: "Vid"
+- family-names: "Leskovec"
+  given-names: "Jure"
+- family-names: "Fey"
+  given-names: "Matthias"
 date-released: 2023-10-24
 license: MIT
 url: "https://github.com/pyg-team/pytorch-frame"

README.md

@@ -264,3 +264,15 @@ pip install pytorch_frame
 ```
 
 See [the installation guide](https://pytorch-frame.readthedocs.io/en/latest/get_started/installation.html) for other options.
+
+## Cite
+
+If you use PyTorch Frame in your work, please cite our paper (Bibtex below).
+```
+@article{hu2024pytorch,
+  title={PyTorch Frame: A Modular Framework for Multi-Modal Tabular Learning},
+  author={Hu, Weihua and Yuan, Yiwen and Zhang, Zecheng and Nitta, Akihiro and Cao, Kaidi and Kocijan, Vid and Leskovec, Jure and Fey, Matthias},
+  journal={arXiv preprint arXiv:2404.00776},
+  year={2024}
+}
+```

pyproject.toml

@@ -50,6 +50,7 @@ full=[
     "scikit-learn",
     "xgboost>=1.7.0, <2.0.0",
     "optuna>=3.0.0",
+    "optuna-integration",
     "mpmath==1.3.0",
     "catboost",
     "lightgbm",

torch_frame/testing/image_embedder.py

@@ -22,5 +22,5 @@ def __init__(
         super().__init__()
         self.out_channels = out_channels
 
-    def forward_embed(self, images: list[Image]) -> Tensor:
+    def forward_embed(self, images: list[Image.Image]) -> Tensor:
         return torch.rand(len(images), self.out_channels)