mirrors/SimCLR
1
0
Fork 0
mirror of https://github.com/sthalles/SimCLR.git synced 2025-06-03 15:03:00 +08:00
Code Issues Packages Projects Releases Wiki Activity

added tensorboard support

Browse Source
This commit is contained in:
Thalles 2020-02-20 14:47:16 -03:00
parent 0d7ff950cb
commit b2d4200b5a
1 changed files with 10 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

27
FeatureEvaluation.ipynb
View File

@ -195,7 +195,7 @@
}, },
{ {
"cell_type": "code", "cell_type": "code",
"execution_count": 18, "execution_count": 23,
"metadata": {}, "metadata": {},
"outputs": [ "outputs": [
{ {
@ -298,7 +298,7 @@
"<All keys matched successfully>" "<All keys matched successfully>"
] ]
}, },
"execution_count": 18, "execution_count": 23,
"metadata": {}, "metadata": {},
"output_type": "execute_result" "output_type": "execute_result"
} }
@ -323,7 +323,7 @@
}, },
{ {
"cell_type": "code", "cell_type": "code",
"execution_count": 19, "execution_count": 24,
"metadata": {}, "metadata": {},
"outputs": [ "outputs": [
{ {
@ -350,7 +350,7 @@
}, },
{ {
"cell_type": "code", "cell_type": "code",
"execution_count": 20, "execution_count": 25,
"metadata": {}, "metadata": {},
"outputs": [ "outputs": [
{ {
@ -377,7 +377,7 @@
}, },
{ {
"cell_type": "code", "cell_type": "code",
"execution_count": 21, "execution_count": 26,
"metadata": {}, "metadata": {},
"outputs": [ "outputs": [
{ {
@ -402,7 +402,7 @@
}, },
{ {
"cell_type": "code", "cell_type": "code",
"execution_count": 22, "execution_count": 27,
"metadata": {}, "metadata": {},
"outputs": [ "outputs": [
{ {
@ -410,8 +410,8 @@
"output_type": "stream", "output_type": "stream",
"text": [ "text": [
"SimCLR feature evaluation\n", "SimCLR feature evaluation\n",
"Train score: 0.7444\n", "Train score: 0.7852\n",
"Test score: 0.62625\n" "Test score: 0.641625\n"
] ]
} }
], ],
@ -420,16 +420,9 @@
"print(\"Train score:\", clf.score(X_train_feature, y_train))\n", "print(\"Train score:\", clf.score(X_train_feature, y_train))\n",
"print(\"Test score:\", clf.score(X_test_feature, y_test))\n", "print(\"Test score:\", clf.score(X_test_feature, y_test))\n",
"# SimCLR feature evaluation\n", "# SimCLR feature evaluation\n",
"# Train score: 0.7444\n", "# Train score: 0.7852\n",
"# Test score: 0.62625" "# Test score: 0.641625"
] ]
},
{
"cell_type": "code",
"execution_count": null,
"metadata": {},
"outputs": [],
"source": []
} }
], ],
"metadata": { "metadata": {