scikit-learn/transformers-imdb

Model description

This is a logistic regression model trained with GPT-2 embeddings on imdb dataset. The notebook to generate this model is in this repository and in this kaggle link.

Intended uses & limitations

This model is trained for educational purposes.

Training Procedure

Hyperparameters

The model is trained with below hyperparameters.

<details> <summary> Click to expand </summary>

</details>

Model Plot

The model plot is below.

<style>#sk-c17251a9-68a0-4b34-a80a-a89592893866 {color: black;background-color: white;}#sk-c17251a9-68a0-4b34-a80a-a89592893866 pre{padding: 0;}#sk-c17251a9-68a0-4b34-a80a-a89592893866 div.sk-toggleable {background-color: white;}#sk-c17251a9-68a0-4b34-a80a-a89592893866 label.sk-toggleable_label {cursor: pointer;display: block;width: 100%;margin-bottom: 0;padding: 0.3em;box-sizing: border-box;text-align: center;}#sk-c17251a9-68a0-4b34-a80a-a89592893866 label.sk-toggleablelabel-arrow:before {content: "▸";float: left;margin-right: 0.25em;color: #696969;}#sk-c17251a9-68a0-4b34-a80a-a89592893866 label.sk-toggleablelabel-arrow:hover:before {color: black;}#sk-c17251a9-68a0-4b34-a80a-a89592893866 div.sk-estimator:hover label.sk-toggleablelabel-arrow:before {color: black;}#sk-c17251a9-68a0-4b34-a80a-a89592893866 div.sk-toggleablecontent {max-height: 0;max-width: 0;overflow: hidden;text-align: left;background-color: #f0f8ff;}#sk-c17251a9-68a0-4b34-a80a-a89592893866 div.sk-toggleablecontent pre {margin: 0.2em;color: black;border-radius: 0.25em;background-color: #f0f8ff;}#sk-c17251a9-68a0-4b34-a80a-a89592893866 input.sk-toggleablecontrol:checked~div.sk-toggleablecontent {max-height: 200px;max-width: 100%;overflow: auto;}#sk-c17251a9-68a0-4b34-a80a-a89592893866 input.sk-toggleablecontrol:checked~label.sk-toggleablelabel-arrow:before {content: "▾";}#sk-c17251a9-68a0-4b34-a80a-a89592893866 div.sk-estimator input.sk-toggleablecontrol:checked~label.sk-toggleablelabel {background-color: #d4ebff;}#sk-c17251a9-68a0-4b34-a80a-a89592893866 div.sk-label input.sk-toggleablecontrol:checked~label.sk-toggleablelabel {background-color: #d4ebff;}#sk-c17251a9-68a0-4b34-a80a-a89592893866 input.sk-hidden--visually {border: 0;clip: rect(1px 1px 1px 1px);clip: rect(1px, 1px, 1px, 1px);height: 1px;margin: -1px;overflow: hidden;padding: 0;position: absolute;width: 1px;}#sk-c17251a9-68a0-4b34-a80a-a89592893866 div.sk-estimator {font-family: monospace;background-color: #f0f8ff;border: 1px dotted black;border-radius: 0.25em;box-sizing: border-box;margin-bottom: 0.5em;}#sk-c17251a9-68a0-4b34-a80a-a89592893866 div.sk-estimator:hover {background-color: #d4ebff;}#sk-c17251a9-68a0-4b34-a80a-a89592893866 div.sk-parallel-item::after {content: "";width: 100%;border-bottom: 1px solid gray;flex-grow: 1;}#sk-c17251a9-68a0-4b34-a80a-a89592893866 div.sk-label:hover label.sk-toggleablelabel {background-color: #d4ebff;}#sk-c17251a9-68a0-4b34-a80a-a89592893866 div.sk-serial::before {content: "";position: absolute;border-left: 1px solid gray;box-sizing: border-box;top: 2em;bottom: 0;left: 50%;}#sk-c17251a9-68a0-4b34-a80a-a89592893866 div.sk-serial {display: flex;flex-direction: column;align-items: center;background-color: white;padding-right: 0.2em;padding-left: 0.2em;}#sk-c17251a9-68a0-4b34-a80a-a89592893866 div.sk-item {z-index: 1;}#sk-c17251a9-68a0-4b34-a80a-a89592893866 div.sk-parallel {display: flex;align-items: stretch;justify-content: center;background-color: white;}#sk-c17251a9-68a0-4b34-a80a-a89592893866 div.sk-parallel::before {content: "";position: absolute;border-left: 1px solid gray;box-sizing: border-box;top: 2em;bottom: 0;left: 50%;}#sk-c17251a9-68a0-4b34-a80a-a89592893866 div.sk-parallel-item {display: flex;flex-direction: column;position: relative;background-color: white;}#sk-c17251a9-68a0-4b34-a80a-a89592893866 div.sk-parallel-item:first-child::after {align-self: flex-end;width: 50%;}#sk-c17251a9-68a0-4b34-a80a-a89592893866 div.sk-parallel-item:last-child::after {align-self: flex-start;width: 50%;}#sk-c17251a9-68a0-4b34-a80a-a89592893866 div.sk-parallel-item:only-child::after {width: 0;}#sk-c17251a9-68a0-4b34-a80a-a89592893866 div.sk-dashed-wrapped {border: 1px dashed gray;margin: 0 0.4em 0.5em 0.4em;box-sizing: border-box;padding-bottom: 0.4em;background-color: white;position: relative;}#sk-c17251a9-68a0-4b34-a80a-a89592893866 div.sk-label label {font-family: monospace;font-weight: bold;background-color: white;display: inline-block;line-height: 1.2em;}#sk-c17251a9-68a0-4b34-a80a-a89592893866 div.sk-label-container {position: relative;z-index: 2;text-align: center;}#sk-c17251a9-68a0-4b34-a80a-a89592893866 div.sk-container {/* jupyter's `normalize.less` sets `[hidden] { display: none; }` but bootstrap.min.css set `[hidden] { display: none !important; }` so we also need the `!important` here to be able to override the default hidden behavior on the sphinx rendered scikit-learn.org. See: https://github.com/scikit-learn/scikit-learn/issues/21755 */display: inline-block !important;position: relative;}#sk-c17251a9-68a0-4b34-a80a-a89592893866 div.sk-text-repr-fallback {display: none;}</style><div id="sk-c17251a9-68a0-4b34-a80a-a89592893866" class="sk-top-container"><div class="sk-text-repr-fallback"><pre>Pipeline(steps=[(&#x27;embedding&#x27;,HFTransformersLanguage(modelnameorpath=&#x27;facebook/bart-base&#x27;)),(&#x27;model&#x27;, LogisticRegression())])</pre><b>Please rerun this cell to show the HTML repr or trust the notebook.</b></div><div class="sk-container" hidden><div class="sk-item sk-dashed-wrapped"><div class="sk-label-container"><div class="sk-label sk-toggleable"><input class="sk-toggleable_control sk-hidden--visually" id="cae9454b-9d5c-424d-bbf8-8256c92c6733" type="checkbox" ><label for="cae9454b-9d5c-424d-bbf8-8256c92c6733" class="sk-toggleablelabel sk-toggleablelabel-arrow">Pipeline</label><div class="sk-toggleablecontent"><pre>Pipeline(steps=[(&#x27;embedding&#x27;,HFTransformersLanguage(modelnameorpath=&#x27;facebook/bart-base&#x27;)),(&#x27;model&#x27;, LogisticRegression())])</pre></div></div></div><div class="sk-serial"><div class="sk-item"><div class="sk-estimator sk-toggleable"><input class="sk-toggleable_control sk-hidden--visually" id="0e6ffcb1-dfbc-44ef-9a7c-c15d496369c7" type="checkbox" ><label for="0e6ffcb1-dfbc-44ef-9a7c-c15d496369c7" class="sk-toggleablelabel sk-toggleablelabel-arrow">HFTransformersLanguage</label><div class="sk-toggleablecontent"><pre>HFTransformersLanguage(modelnameorpath=&#x27;facebook/bart-base&#x27;)</pre></div></div></div><div class="sk-item"><div class="sk-estimator sk-toggleable"><input class="sk-toggleable_control sk-hidden--visually" id="b8eb1956-cf67-40f5-96f2-c2f0a0a41704" type="checkbox" ><label for="b8eb1956-cf67-40f5-96f2-c2f0a0a41704" class="sk-toggleablelabel sk-toggleablelabel-arrow">LogisticRegression</label><div class="sk-toggleable_content"><pre>LogisticRegression()</pre></div></div></div></div></div></div></div>

Evaluation Results

You can find the details about evaluation process and the evaluation results.

How to Get Started with the Model

Use the code below to get started with the model.

<details> <summary> Click to expand </summary>

[More Information Needed]

</details>

# Additional Content

## Confusion matrix

![Confusion matrix](confusion_matrix.png)