Skip to content

Commit

Permalink
Update README.md
Browse files Browse the repository at this point in the history
  • Loading branch information
escrogar authored Nov 8, 2021
1 parent ec9b87d commit eb26eae
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion README.md
Original file line number Diff line number Diff line change
Expand Up @@ -55,7 +55,7 @@ The approach was benchmarked against embeddings from a non fine-tuned XLM-Robert
## Usage
The input is needed in a tsv file, containing two necessary columns: "text" for the text itself, and "topik" for the numeric category labels. The code provides a JSON file for the results compiled in a dictionary, and another one for the optimized parameters.

The code and approach are still under refinement. It was prepared to run in a GPU-supported CUDA environment, but another (classless) version is provided for easy usage and debugging in a free Google Colab GPU-supported notebook. Minor modifications might be needed to adapt to own directory and data structure.
The code and approach are still under refinement. It was prepared to run in a GPU-supported CUDA environment. A notebook version is provided for easy usage and debugging in a free Google Colab GPU-supported notebook. Minor modifications might be needed to adapt to own directory and data structure.

## Packages required:
pandas, torch, transformers, numpy, json, sklearn, google.drive (optional)
Expand Down

0 comments on commit eb26eae

Please sign in to comment.